js实现 三数之和 LeetCode
题目描述
我感觉这篇文章写的特别好,把我遇到的坑都写到了。我只是为了学习,在这一记录一遍
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?
请你找出所有和为 0 且不重复的三元组
注意:答案中不可以包含重复的三元组。
解题思路
1.首先将数据进行排序,记住nums[i],从i后面找两个数之和是-nums[i]这样三个数的和就是0了。先声明一个数组,储存最后的结果
2.设置target = -nums[i],再设置两个标识left和right,遍历i之后的数组元素,计算left和right两个位置的元素之和sum,若
sum>target,则指针right向前移动,若sum<target 则指针left向后移动,最后就是相等的情况了,相等则将这3个数组成的数组
保存进res中。
3.增加一些限制条件,首先nums[i]若大于0那肯定找不到满足条件的解了所以可以作为终止条件。
代码实现:
var threeSum = function(nums) {
var res = []
nums = nums.sort((a,b) => a-b)// 排序
for(var i=0;i<nums.length-2;i++){
// 去重复,如果这个数和前面的数一样,则没有必要进行下面的操作 因为得到的数组是相同的
if(nums[i] == nums[i-1] ){
continue
}
var target = -nums[i]
var left = i+1
var right = nums.length - 1
while(left<right){
var sum = nums[left] + nums[right]
if(sum > target)
right--
else if(sum < target)
left++
else{
res.push([nums[i],nums[left],nums[right]])
// 去重复,若后面有与nums[left]重复的数字,则得到的数组是相同的,所以在这里去重复
while(nums[left] == nums[left+1]){
left++
}
left++
// 去重复
while(nums[right] == nums[right-1]){
right--
}
right--
}
}
}
return res
};
遇到的坑
1.sort()
在javascript中sort是按照字符顺序来排序的,例如:
var arr=[12,23,7,8,31]
console.log(arr.sort())//结果是[ 12, 23, 31, 7, 8 ]
如果需要按照数值来排序,则需要添加一个比较函数来做参数
升序:
var arr=[12,23,7,8,31]
console.log(arr.sort((a,b)=>a-b))//输出是[ 7, 8, 12, 23, 31 ]
降序:
var arr=[12,23,7,8,31]
console.log(arr.sort((a,b)=>b-a))//输出是[ 31, 23, 12, 8, 7 ]
2.去重遇到的坑
一开始,去重部分我是这么写的
if(nums[i] == nums[i+1] ){
continue
}
但是得到的结果是[[-1,0,1]]
这是为什么呢?
如果是 [0,0,0,…],[-2,-2,4,…]类似这种情况,连续3个数刚好就是一组解的时候,然后这组解有重复元素时,上面 的遍历方式就把这些解给过滤掉了,所以应该取第一个元素计算,即先计算再判断是否与之前的重复,而不是先判断是否会重复再选择计算最后一个。所以去重部分的代码应该是:
if(nums[i] == nums[i-1] ){
continue
}
而且left和right部分也应该去重
while(nums[left] == nums[left+1]){
left++
}
while(nums[right] == nums[right-1]){
right--
}