美文网首页
2006. 差的绝对值为 K 的数对数目

2006. 差的绝对值为 K 的数对数目

作者: Abeants | 来源:发表于2022-02-09 22:23 被阅读0次

给你一个整数数组 nums 和一个整数 k ,请你返回数对 (i, j) 的数目,满足 i < j 且 |nums[i] - nums[j]| == k 。

|x| 的值定义为:
如果 x >= 0 ,那么值为 x 。
如果 x < 0 ,那么值为 -x 。

示例 1:
输入:nums = [1,2,2,1], k = 1
输出:4
解释:差的绝对值为 1 的数对为:
[1,2,2,1]
[1,2,2,1]
[1,2,2,1]
[1,2,2,1]

示例 2:
输入:nums = [1,3], k = 3
输出:0
解释:没有任何数对差的绝对值为 3 。

示例 3:
输入:nums = [3,2,1,5,4], k = 2
输出:3
解释:差的绝对值为 2 的数对为:
[3,2,1,5,4]
[3,2,1,5,4]
[3,2,1,5,4]

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/count-number-of-pairs-with-absolute-difference-k
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解题思路及方法

可以暴力法:

class Solution {
    public int countKDifference(int[] nums, int k) { 
        // 暴力法
        int res = 0;

        for (int i = 0; i < nums.length - 1; i++) {
            for (int j = i + 1; j < nums.length; j++) {
                if (Math.abs(nums[j] - nums[i]) == k) res++;
            }
        }

        return res;
    }
}

也可以计数,记录数组中每个数出现的次数,然后统计满足 |nums[i] - nums[j]| == k的个数,即counts[i] * counts[i + k]不等于0,即counts[i]和counts[i + k]的出现次数均至少为1,相乘是因为可以,如果counts[i]出现m次,counts[i + k]出现n次,满足情况总共为m*n次。

class Solution {
    public int countKDifference(int[] nums, int k) {
        // 计数
        int[] counts = new int[101];

        for (int i : nums) {
            counts[i]++;
        }

        int res = 0;
        for (int i = 1; i + k < 101; i++) {
            res += counts[i] * counts[i + k];
        }

        return res;
    }
}

结果如下:

相关文章

网友评论

      本文标题:2006. 差的绝对值为 K 的数对数目

      本文链接:https://www.haomeiwen.com/subject/gjugkrtx.html