961. 重复N次的元素(Python)
2019-06-01 本文已影响0人
玖月晴
更多精彩内容,请关注【力扣简单题】。
题目
难度:★☆☆☆☆
类型:数组
在大小为 2N 的数组 A 中有 N+1 个不同的元素,其中有一个元素重复了 N 次。
返回重复了 N 次的那个元素。
提示
4 <= A.length <= 10000
0 <= A[i] < 10000
A.length 为偶数
示例
示例 1
输入:[1,2,3,3]
输出:3
示例 2
输入:[2,1,2,5,3,2]
输出:2
示例 3
输入:[5,1,5,2,5,3,5,4]
输出:5
解答
方案1:频率统计
统计每个词出现的次数,找出出现次数为N的元素。
class Solution:
def repeatedNTimes(self, A):
"""
:param A: List[int]
:return: int
"""
from collections import Counter
return [num for num, count in Counter(A).items() if count == len(A) // 2][0]
方案2:考虑数量
有一个元素出现N次,且其他N个元素各不相同,利用这个条件,只可能有两种情况:
出现N次的数有两个相邻,则相邻的数即为所求;
出现N次的数有两两均不相邻,则连续4个数之中一定有两个该数字,找到即可。
class Solution:
def repeatedNTimes(self, A):
"""
:param A: List[int]
:return: int
"""
for i in range(len(A)-1):
if A[i] == A[i+1]:
return A[i]
return A[0] if A[0] == A[2] or A[0] == A[3] else A[1]
方案3:使用集合
我们将输入数组A中的元素逐个放入集合tmp中,若在tmp中发现重复的元素,则返回这个元素。由于重复元素的数量很多,占到输入数组的一半,所以运行效率会很高。
class Solution:
def repeatedNTimes(self, A):
"""
:param A: List[int]
:return: int
"""
tmp = set()
for c in A:
if c in tmp:
return c
else:
tmp.add(c)
如有疑问或建议,欢迎评论区留言~