相差为k的数对数量

最新推荐文章于 2026-01-02 10:17:02 发布

原创最新推荐文章于 2026-01-02 10:17:02 发布 · 190 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #数据结构 #java

算法专栏收录该内容

19 篇文章

订阅专栏

给定一个整数数组nums和一个整数k，求出数组中所有满足i<j且|nums[i]-nums[j]|=k的数对(i,j)的数量。解决方案是使用哈希表统计每个数字出现的次数，然后枚举数组，检查加减k后的数字是否在哈希表中，累加对应计数并更新哈希表。

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

|x| 的值定义为：

如果 x >= 0 ，那么值为 x 。
如果 x < 0 ，那么值为 -x 。

1 <= nums.length,k <= 100000，数组中每个数字的范围在[1，20000]之间

思路：使用哈希表存储每个出现的次数，枚举原数组，将每个数字加上或减去 k，如果哈希表中存在加上或减去k之后的数，答案累加上该数出现的次数。+ - 操作后，将枚举的数字在哈希表中减去 1 ，避免后续出现重复。

class Solution {
    public int countKDifference(int[] nums, int k) {
         //这里直接开个新的数组统计nums中元素出现的次数，但后续的枚举需要判断边界	
         int[] cnt = new int[20002];
         int n = nums.length;
         for(int i = 0; i < n ; i ++){
             cnt[nums[i]]++;
         }
         int res = 0;
         for(int i = 0; i < n ; i ++){
             if(cnt[nums[i] + k] > 0){
                res += cnt[nums[i] + k];
             }
             if(nums[i] - k >= 0 && cnt[nums[i] - k ] > 0){
                 res += cnt[nums[i] - k ];
             }
             cnt[nums[i]]--;
         }
         return res;
    }
}