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& nums) {
    std::unordered_map  mapCount;
    int res = 0;
    if(nums.empty())
        return res;
    for (auto i = 0; i < nums.size(); ++i) 
    {
        std::unordered_map ::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;

    }
};

你可能感兴趣的:(C++,leetcode,c++)