算法提高之LeetCode刷题算法

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)

如有疑问或建议,欢迎评论区留言~

上一篇下一篇

猜你喜欢

热点阅读