给出一个数组,在其中找到三个数相加使得结果为0?
思路:
从数组中找到第一个数,然后用0减去这个数,得到一个sum,然后在剩下的几个数中从两头开始遍历找两个数相加的和为sum,如果第一次找到了这两个数,那么分别把这两个数向中间递进,如果相同则去掉;如果没有找到把这两头的数换掉。最后如果第一个数确定后找不到两个数使三个和为0,则说明可以把第二个数下标加一个,第三个数不变;或者第二个数不变,第三个数下边减一。
具体代码
class Solution {
public List<List<Integer>> threeSum(int[] num) {
//首先对数组进行排序,从大到小
Arrays.sort(num);
//new一个链表,链表的元素是列表
List<List<Integer>> res=new LinkedList<>();
//i从第一个开始,一直到倒数第三个,因为要找出三个数相加的和
for(int i=0;i<num.length-2;i++){
//这里不能写成num[i]!=num[i+1],否则如果两个元素相同的话会直接跳过去
if(i==0||(i>0&&num[i-1]!=num[i])){
//两端进行逼近,下标分别为j,k,以及这两个数相加的和
int j=i+1;
int k=num.length-1;
int sum=0-num[i];
while(j<k){
if(num[j]+num[k]==sum){
res.add(Arrays.asList(num[i],num[j],num[k]));
//如果j<k并且num[j]=num[j+1],那么j就加1,不考虑后一个,因为是相同的输出
while(j<k&&num[j+1]==num[j]) j++;
while(j<k&&num[k-1]==num[k]) k--;
//如果j和k相邻的都不相等,那么j和k分别++和--向中间靠近
j++;k--;
}
//如果小于sum,那么就应该把其中一个数增大,因为是顺序排过的,所以j++;
else if(num[j]+num[k]<sum){
j++;
}
else{
k--;
}
}
}
}
return res;
}
}