题目描述
给你一个整数数组 nums
。
如果一组数字 (i,j)
满足 nums[i] == nums[j]
且 i < j
,就可以认为这是一组 好数对 。
返回好数对的数目。
样例1
输入:nums = [1,2,3,1,1,3]
输出:4
解释:有 4 组好数对,分别是 (0,3), (0,4), (3,4), (2,5) ,下标从 0 开始
样例2
输入:nums = [1,1,1,1]
输出:6
解释:数组中的每组数字都是好数对
样例3
输入:nums = [1,2,3]
输出:0
限制
- $1 \leq nums.length \leq 100$
- $1 \leq nums[i] \leq 100$
算法
(枚举) $O(n)$
- 分别记录不同的数的出现次数
- 每一组数中任意选择两个数可以构成一个好数对
- 枚举所有不同的数求组合数即可
时间复杂度
原数组和$hash$表都只会遍历一次
C++ 代码
class Solution {
public:
int numIdenticalPairs(vector<int>& nums) {
unordered_map<int, int> cnt;
for (auto x: nums) cnt[x] ++ ;
int ret = 0;
for (auto x: cnt) ret += x.second * (x.second - 1) / 2;
return ret;
}
};