【算法】1512. 好数对的数目(java / c / c++ / python / go / rust)

举报
二当家的白帽子 发表于 2022/02/08 12:23:21 2022/02/08
【摘要】 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...

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://bbs.huaweicloud.com/community/usersnew/id_1628396583336561 博客原创~


【版权声明】本文为华为云社区用户原创内容,转载时必须标注文章的来源(华为云社区)、文章链接、文章作者等基本信息, 否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。