题目及要求
给你一个整数数组 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 <= nums.length <= 100
1 <= nums[i] <= 100
暴力算法
思路:双循环每个数字都遍历过去,符合要求ans自增,最后返回ans即好数对的数目
class Solution {
public:
int numIdenticalPairs(vector<int>& nums) {
int n=nums.size();
int ans=0;
for(int i=0;i<n;++i){
for(int j=i+1;j<n;++j){
if(nums[i]==nums[j]&&i<j)++ans;
}
}
return ans;
}
};
哈希算法
思路:将出现的次数存入哈希然后去判断并不断更新goodpairs的值,最后返回即可
class Solution {
public:
int numIdenticalPairs(vector<int>& nums) {
unordered_map<int, int> hash;
int goodPairs = 0;
for (int num : nums) {
// 如果 countMap 中已经有了当前数字,则更新好数对的数量
if (hash.find(num) != hash.end()) {
goodPairs += hash[num];
}
// 更新当前数字的出现次数
hash[num]++;
}
return goodPairs;
}
};
在main内使用
int main() {
Solution solution;
vector<int> nums = {1, 2, 3, 1, 1, 2, 2};
int result = solution.numIdenticalPairs(nums);
cout << "好数对: " << result <<endl;
return 0;
}