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/
非常感谢你阅读本文~
放弃不难,但坚持一定很酷~
希望我们大家都能每天进步一点点~
本文由 二当家的白帽子:https://developer.aliyun.com/profile/sqd6avc7qgj7y 博客原创~