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

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
    }
}

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


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


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

上一篇:云博士智能诊断功能全新升级


下一篇:vim快捷键笔记【原创】