【题目】
给你一个整数数组 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
【代码】
暴力法:
class Solution {
public:
int numIdenticalPairs(vector<int>& nums) {
int cnt=0;
for(int i=0;i<nums.size();i++){
for(int j=i+1;j<nums.size();j++){
if(nums[i]==nums[j])
cnt++;
}
}
return cnt;
}
};
机智版本:前n项求和公式,Sn=(a1+an)*n/2
class Solution {
public:
int numIdenticalPairs(vector<int>& nums) {
map<int,int> m;
int cnt=0;
for(auto x:nums)
m[x]++;
for(int i=0;i<m.size();i++)
if(m[i]>1)
cnt+=m[i]*(m[i]-1)/2;
return cnt;
}
};