持续创作,加速成长!这是我参与「掘金日新计划 · 6 月更文挑战」的第29天,点击查看活动详情
编程世界总是离不了算法
最近在看框架源码时,会有很多算法的实现逻辑,有时候会感到吃力
于是决定蹭着假期,加强算法和数据结构相关的知识
那怎么提升呢?
其实我知道算法这东西没有捷径,多写多练才能提升,于是我开启我的LeetCode
刷题之旅
第一阶段目标是:200
道,每天1
到2
篇
为了不乱,本系列文章目录分为三部分:
- 今日题目:xxx
- 我的思路
- 代码实现
532. 数组中的 k-diff 数对
给你一个整数数组 nums 和一个整数 k,请你在数组中找出 不同的 k-diff 数对,并返回不同的 k-diff 数对 的数目。
k-diff 数对定义为一个整数对 (nums[i], nums[j]) ,并满足下述全部条件:
0 <= i, j < nums.length i != j nums[i] - nums[j] == k 注意,|val| 表示 val 的绝对值。
示例 1:
输入:nums = [3, 1, 4, 1, 5], k = 2
输出:2
解释:数组中有两个 2-diff 数对, (1, 3) 和 (3, 5)。
尽管数组中有两个 1 ,但我们只应返回不同的数对的数量。
示例 2:
输入:nums = [1, 2, 3, 4, 5], k = 1
输出:4
解释:数组中有四个 1-diff 数对, (1, 2), (2, 3), (3, 4) 和 (4, 5) 。
示例 3:
输入:nums = [1, 3, 1, 5, 4], k = 0
输出:1
解释:数组中只有一个 0-diff 数对,(1, 1) 。
提示:
1 <= nums.length <= 104
-107 <= nums[i] <= 107
0 <= k <= 107
来源:力扣(LeetCode) 链接:leetcode.cn/problems/k-… 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
我的思路
1.本题解使用了二分查找的方法。
2.分析题目,本题要求找出nums[i] - nums[j] === k的不同数对的数目。分析这个公式,我们可以将其变形nums[i] - k === nums[j],我们可以通过遍历nums对每一个下标i找满足公式的nums[j]。这样我们就把题目转化成了找target的题目,这种时候我们很容易联想到二分查找,并且由于是找数对,所以原数组的顺序并不重要,所以我们可以将nums排序,因此我们可以使用二分查找。
3.进行二分查找时,由于对于每一次遍历的下标i,它实际上是数对的右边界,因此应该将i作为二分查找的右边界。并且注意while的条件(这里可以记为一个模版,当右边界取得到的时候就使用left < right,也可以通过分析得出,因此如果取left <= right,则二分查找会陷入死循环)。
4.最后要小心题目的找不同的数对(数值不同),因此我们可以使用一个哈希集合储存找到的满足条件的nums[j](因为nums[j]一确定根据等式,那么nums[i]也就确定了,所以不能有相同的nums[j])。
代码实现
/**
* @param {number[]} nums
* @param {number} k
* @return {number}
*/
var findPairs = function(nums, k) {
nums.sort((a,b) => a-b);
let set = new Set();
for(let i = 0;i < nums.length;i++)
{
const j = binarySearch(nums,i,nums[i] - k);
if(j !== null)
{
set.add(nums[j])
}
}
return set.size;
};
function binarySearch(nums,end,target)
{
let left = 0;
let right = end;
while(left < right)
{
let mid = Math.floor((left + right) / 2);
if(nums[mid] > target)
{
right = mid;
}
else if(nums[mid] < target){
left = mid + 1;
}
else{
return mid;
}
}
return null;
}
总结
实现方式其实有很多,这里仅供参考~
由于刚开始刷题,也不知道从哪里刷好,如果前辈们有好的建议,希望不吝赐教,感谢🌹