leetcode 1512. 好数对的数目

leetcode链接：leetcode 1512. 好数对的数目

题目描述

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

思路

若某数字 k 在数组中出现 n 次，那么满足 nums[i] = nums[j] = k的排列组合应该为 A(n,2)，也就是 n(n-1)。但是考虑到 i < j。那么在所有排列中，只有一半的排列满足要求，即 n(n-1)/2。

AC代码

class Solution {
public: 
    int numIdenticalPairs(vector<int>& nums) {
    std::unordered_map <int, int> mapCount;
    int res = 0;
    if(nums.empty())
        return res;
    for (auto i = 0; i < nums.size(); ++i) 
    {
        std::unordered_map <int, int>::iterator iterator(mapCount.find(nums[i]));
        if (iterator != mapCount.end()) {
            iterator->second++;
        }
        else {
            mapCount[nums[i]] = 1;
        }
    }
    
    for (auto const& ele : mapCount)
    {
        res += (ele.second * (ele.second - 1)) / 2;        
    }
    return res;

    }
};