算法学习1512. 好数对的数目（java / c / c++ / python / go / rust）

Posted 2021-11-10 二当家的白帽子

非常感谢你阅读本文~
欢迎【👍点赞】【⭐收藏】【📝评论】~
放弃不难，但坚持一定很酷~
希望我们大家都能每天进步一点点~
本文由 二当家的白帽子 https://le-yi.blog.csdn.net/ 博客原创~

---

文章目录

• 1512. 好数对的数目：
• 样例 1
• 样例 2
• 样例 3
• 提示
• 分析
• 题解
• • java
  • c
  • c++
  • python
  • go
  • rust
• 原题传送门：https://leetcode-cn.com/problems/number-of-good-pairs/

---

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

---

分析

• 双层循环，暴力统计是最普通的方法。
• 事实上，我们可以利用排列组合的知识，先统计每个数字有多少个，然后再看从中任意选2个，有多少种选法。
• 我们也可以一边计数，一边计算结果，如果一个数字是第一次出现，没法凑出对，如果之前已经统计有这个数字，那么其实就是当前位置和之前每一个位置组合一次，所以好数对增加的数量就是当前这个数字之前出现的次数。

---

题解

java

class Solution {
    public int numIdenticalPairs(int[] nums) {
        int ans = 0;
        
        int[] counter = new int[101];
        for (int n : nums) {
           ans += counter[n]++;
        }

        return ans;
    }
}

---

c

int numIdenticalPairs(int* nums, int numsSize){
    int ans = 0;

    int counter[101];
    memset(counter, 0, sizeof(counter));
    for (int i = 0; i < numsSize; ++i) {
        ans += counter[nums[i]]++;
    }

    return ans;
}

---

c++

class Solution {
public:
    int numIdenticalPairs(vector<int>& nums) {
        int ans = 0;

        int counter[101];
        memset(counter, 0, sizeof(counter));
        for (auto n : nums) {
            ans += counter[n]++;
        }

        return ans;
    }
};

---

python

class Solution:
    def numIdenticalPairs(self, nums: List[int]) -> int:
        ans = 0

        counter = [0] * 101
        for n in nums:
            ans += counter[n]
            counter[n] += 1

        return ans

---

go

func numIdenticalPairs(nums []int) int {
    ans := 0

	counter := [101]int{}
	for _, n := range nums {
		ans += counter[n]
		counter[n]++
	}

	return ans
}

---

rust

impl Solution {
    pub fn num_identical_pairs(nums: Vec<i32>) -> i32 {
        let mut ans = 0;
    
        let mut counter = vec![0; 101];

        nums.iter().for_each(|n| {
            ans += counter[*n as usize];
            counter[*n as usize] += 1;
        });
        
        ans
    }
}

---

---

原题传送门：https://leetcode-cn.com/problems/number-of-good-pairs/

---