题目:
给定一个包含 n 个整数的数组 nums
,判断 nums
中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
此题为数组题,思前想后觉得用三指针遍历比较靠谱。先将数组排序,使用i,j,k三个指针。i指向最小的数可能的位置,k指向最大的数的位置,j指向中间的数可能的位置。遍历原理核心为每次遍历会遇到得三种情况:
1.nums[i]+nums[j]==-nums[k],此时说明刚好三数加一起等于0,于是此时要把j向右拨一位,但此时nums[i]+nums[j]变大了(nums[j]变大的情况下),所以同时也要把k向左拨。也即此时i++,k--;
2.nums[i]+nums[j]>-nums[k],此时说明nums[i]+nums[j]+nums[k]>0,所以k要向左拨,k--;
3.nums[i]+nums[j]<-nums[k],此时说明nums[i]+nums[j]+nums[k]<0,所以j要向右拨,j++;
当然,遍历的时候肯定会考虑重复等一些的情况,所以要加上邻位判断。比如在遍历第一种情况时,如果nums[j]==nums[j-1],说明此时j已经考虑过了,直接j++即可。
贴上代码。
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
List <List<Integer>> result = new ArrayList<>();
int i;
int j;
int k;
Arrays.sort(nums);
for(i=0;i<nums.length-2;){
j=i+1;
k=nums.length-1;
while(j<k){
if(nums[i]+nums[j]==-nums[k]){
List<Integer> dataput = new ArrayList<>();
dataput.add(nums[i]);
dataput.add(nums[j]);
dataput.add(nums[k]);
result.add(dataput);
j++;
k--;
while(nums[j]==nums[j-1]&&j<k){
j++;
}
while(nums[k]==nums[k+1]&&j<k){
k--;
}
}
else if(nums[i]+nums[j]>-nums[k]){//i+j+k大于0,k要减少
k--;
while(nums[k]==nums[k+1]&&j<k){
k--;
}
}
else if(nums[i]+nums[j]<-nums[k]){//i+j+k大于0,j要增多
j++;
while(nums[j]==nums[j-1]&&j<k){
j++;
}
}
}
i++;
while(nums[i]==nums[i-1]&&i<nums.length-2){
i++;
}
}
return result;
}
}
提交结果如下: