Python Leetcode(961.重复N次的元素)

Python Leetcode(961.重复N次的元素)

在大小为 2N 的数组 A 中有 N+1 个不同的元素，其中有一个元素重复了 N 次。

返回重复了 N 次的那个元素。

示例 1：

输入：[1,2,3,3]
输出：3

示例 2：

输入：[2,1,2,5,3,2]
输出：2

示例 3：

输入：[5,1,5,2,5,3,5,4]
输出：5

提示：

4 <= A.length <= 10000
0 <= A[i] < 10000
A.length 为偶数

Solution1:(统计数组中每个元素出现的次数，其出现次数为数组长度的一半就是要找的数字。)

class Solution(object):
    def repeatedNTimes(self, A):
        """
        :type A: List[int]
        :rtype: int
        """
        count_dict = {}
        for each in A:
            if each not in count_dict:
                count_dict[each] = 1
            else:
                count_dict[each] += 1
        
        for k, v in count_dict.items():
            if v == len(A)/2:
                return k

solution = Solution()
print(solution.repeatedNTimes([1, 2, 3, 3]))

3

solution2:(由题意可知，其他的元素都只出现了一次，那么只要重复的那个元素即为重复N次的元素。)

class Solution(object):
    def repeatedNTimes(self, A):
        """
        :type A: List[int]
        :rtype: int
        """
        count_dict = {}
        for each in A:
            if each not in count_dict:
                count_dict[each] = 1
            else:
                return each

solution = Solution()
print(solution.repeatedNTimes([1, 2, 3, 3]))

3

Solution3:(题解中的方法，首先比较A[0]和A[-1]，若相等则返回A[0]，不相等比较A[1]和A[-1]，相等则返回A[1]；不相等则需要遍历数组，检查是否有相邻的元素相等，有则返回该元素，没有的话直接返回A[0]，这是因为已经比较了A[0]、A[1]、A[-1]三个值都不相等，如果答案不在这三个值里，那么剩下的2N-3个数中的N个重复数字是没有相邻的，那么需要N-1个数字插在其中，数字个数2N-1大于2N-3，不符合题意；另外若最后的答案不是A[0]，则会推出A[1]=A[-1]的结论。)

class Solution(object):
    def repeatedNTimes(self, A):
        """
        :type A: List[int]
        :rtype: int
        """
        if A[0] == A[-1]:
            return A[0]
        elif A[1] == A[-1]:
            return A[1]
        else:
            for i in range(1, len(A)):
                if A[i] == A[i-1]:
                    return A[i]
        return A[0]

solution = Solution()
print(solution.repeatedNTimes([1, 2, 3, 3]))

3