LeetCode刷题（66）～好数对的数目

题目描述

给你一个整数数组 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 开始

  

 

  
1
  
2
  
3
 

示例 2：

输入：nums = [1,1,1,1]
输出：6
解释：数组中的每组数字都是好数对

  

 

  
1
  
2
  
3
 

示例 3：

输入：nums = [1,2,3]
输出：0

  

 

  
1
  
2
 

提示：

• 1 <= nums.length <= 100
• 1 <= nums[i] <= 100

解答 By 海轰

提交代码

 int c(int num) { int sum=1; for(int i=num-2+1;i<=num;++i) sum*=i; return sum/2; } int numIdenticalPairs(vector<int>& nums) { unordered_map<int,int> m; for(int num:nums) ++m[num]; int result=0; for(unordered_map<int,int>::iterator it=m.begin();it!=m.end();++it) { if(it->second>1) result+=c(it->second); } return result; }

  

 

  
1
  
2
  
3
  
4
  
5
  
6
  
7
  
8
  
9
  
10
  
11
  
12
  
13
  
14
  
15
  
16
  
17
  
18
  
19
  
20
 

运行结果

提交代码（优化）

int numIdenticalPairs(vector<int>& nums) { unordered_map<int,int> m; for(int num:nums) ++m[num]; int result=0; for(unordered_map<int,int>::iterator it=m.begin();it!=m.end();++it) { if(it->second>1) result+=(it->second)*(it->second-1)/2; } return result; }

  

 

  
1
  
2
  
3
  
4
  
5
  
6
  
7
  
8
  
9
  
10
  
11
  
12
  
13
 

运行结果

题目来源

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/number-of-good-pairs

文章来源: haihong.blog.csdn.net，作者：海轰Pro，版权归原作者所有，如需转载，请联系作者。

原文链接：haihong.blog.csdn.net/article/details/108057499