Leetcode 1512. 好数对的数目

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

解题思路

暴力求解
使用两层循环求解，第一层for循环用来记录 好数对 的第一个数字nums[i]，第二层for循环用来记录 好数对 的第二个数字 nums[j]，如果 nums[i] == nums[j]，那么 好数对 的数量就加一。这样，时间复杂度就为（O($n^2$)）。

class Solution {
public:
    int numIdenticalPairs(vector<int>& nums) {
        if(nums.size() <= 1){
            return 0;
        }

        int count = 0;
        for(int i=0; i<nums.size(); i++){
            for(int j=nums.size()-1; j>i; j--){
                if(nums[i] == nums[j]){
                    count++;
                }
            }
        }
        return count;
    }
};

哈希表解题

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

class Solution {
public:
    int numIdenticalPairs(vector<int>& nums) {
        unordered_map<int, int> map;
        for(int i=0; i<nums.size(); i++){
            map[nums[i]]++;
        }

        int count = 0;
        for(auto num : map){
            count += (num.second)*(num.second-1)/2;
        }

        return count;
    }
};

根据上面的思路，我们使用两个循环，先记录每个数字出现的次数，然后再计算由这个数字组成的 好数对。下面使用一个for循环完成上面的操作。

class Solution {
public:
    int numIdenticalPairs(vector<int>& nums) {
        unordered_map<int, int> map;
        int count = 0;

        for(int i=0; i<nums.size(); i++){
            if(map.find(nums[i]) != map.end()){
                count += map[nums[i]];
            }
            map[nums[i]] += 1;
        }

        return count;
    }
};

欢迎大家关注我的个人公众号，同样的也是和该博客账号一样，专注分享技术问题，我们一起学习进步