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

提示：

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/n-repeated-element-in-size-2n-array
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

执行结果

代码描述

思路：边扫描A，边将其加入map，并且值作为次数，判断当次数为N时，返回键，即可。

一种方法，边加入，边判断，因为只会存在一个值（次数）为 N，且没有比N大的。

另一种方法，都插入map之后，再去查找次数为N的，并返回键。

// 方法1
class Solution {
public:
    int repeatedNTimes(vector<int>& A) {
        int N = A.size()/2;
        map<int, int> m;
        for(int i = 0; i < A.size(); ++i)
        {
            m[A[i]]++;
            if(m[A[i]] == N)
                return A[i];
        }
        return 0;
    }
};


// 方法2
class Solution {
public:
    int repeatedNTimes(vector<int>& A) {
        int N = A.size()/2;
        map<int, int> m;
        for(int i = 0; i < A.size(); ++i)
        {
            m[A[i]]++;
        }
        
        map<int, int>::iterator it = m.begin();
        for(; it != m.end(); ++it)
        {
            if(it->second == N)
                break;
        }
        return it->first;
    }
};