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 开始

示例 2：

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

示例 3：

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

提示：

• 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;

    }

运行结果

提交代码（优化）

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;

    }

运行结果

题目来源

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