LeetCode之好数对的数目(一千五百一十二)

目录

 

题目

解题

方法一、暴力法

方法二、哈希法


题目

(原题链接:https://leetcode-cn.com/problems/number-of-good-pairs/

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

解题

方法一、暴力法

分析:根据题意直接循环对比 nums[i] == nums[j] 的计数情况即可,非常简单。

代码:(C++)

class Solution {
public:
    int numIdenticalPairs(vector<int>& nums) {
        int count=0;
        for(int i=0;i<nums.size();i++) {
            for(int j=i+1;j<nums.size();j++) {
                if(nums[i] == nums[j]){
                    count++;
                }
            }
        }
        return count;
    }
};

时间复杂度:O(n^2)。

空间复杂度:O(1)。

执行结果:

方法二、哈希法

分析:利用哈希map的特性,通过空间换时间的方式,减小时间复杂度。以元素值为key,索引值为value构建哈希map,并在遍历数组的同时,查询哈希map中是否存在相同的元素。原理也非常简单,可以直接看代码。

代码:(C++)

class Solution {
public:
    int numIdenticalPairs(vector<int>& nums) {
        unordered_map<int, int> m;
        int count = 0;
        for ( auto num : nums ) {
            if ( m.find(num) != m.end() ) {
                count += m[num];
            }
            ++m[num];
        }
        return count;
    }
};

时间复杂度:O(n)。

空间复杂度:O(n)。

执行结果:

如果有疑问,欢迎评论留言或者私信沟通!  

猜你喜欢

转载自blog.csdn.net/liuzehn/article/details/108115685