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 开始

 

方法1：暴力法

int numIdenticalPairs(int* nums, int numsSize){
    int i, j;
    int ret = 0;
    
    for(i = 0; i < numsSize-1; i++) {
        for(j = i+1; j <numsSize; j++) {
            if(nums[i] == nums[j])
                ret++;
        }
    }

    return ret;
}

 

方法2：找到数组里的n个相同数字时，根据排列组合公式，n个相同数字可以凑n(n-1)/2组好数对。

int numIdenticalPairs(int* nums, int numsSize){
    int i;
    int *tmp = malloc(sizeof(int) * 101);
    int ret = 0;

    memset(tmp, 0, 101 * sizeof(int));

    for(i = 0; i < numsSize; i++) {
        tmp[nums[i]]++;
    }

    for(i = 0; i < 101; i++) {
        if(tmp[i] > 1)
            ret += tmp[i] * (tmp[i] - 1) / 2;        
    }

    return ret;
}