相差为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;
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值