题目:
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
- 示例 1:
- 输入:nums = [-1,0,1,2,-1,-4]
- 输出:[[-1,-1,2],[-1,0,1]]
- 示例 2:
- 输入:nums = []
- 输出:[]
- 示例 3:
- 输入:nums = [0]
- 输出:[]
解题思路:
其中第二个问题,从0遍历到length-2是防止数组越界。
其中第三个问题是为了去除重复结果。
给数组排序后可以方便根据三数和的大小进行是start前移(说明值小于零要找个更大的值计算三数和)还是end后移(三数和大于0需要找个更小的数计算三数和)。
代码:
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
ArrayList<List<Integer>> result = new ArrayList<List<Integer>>(); //定义结果存储
Arrays.sort(nums); //将数组排序 方便指针移动找值
for(int i = 0; i< nums.length-2 ;i++){
if(i == 0 || nums[i]!=nums[i-1]){ //判断i是否与前一个数字相等去除重复结果i为0则直接循环
int start = i+1 , end = nums.length-1; //设置两个移动的指针
while(start < end){
if(nums[i]+nums[start]+nums[end] == 0){ //三个数相加为0
result.add(Arrays.asList(nums[i], nums[start], nums[end])); //将结果存入result
start++; //移动start标志,移动后需要判断num[start]是否等于num[start-1]的值
end--; //移动end标志,移动后需要判断num[end]是否等于num[start+1]的值
while(start<end && nums[start] == nums[start - 1]){//去重判断,若重复则继续+1
start++;
}
while (start < end && nums[end] == nums[end + 1]) {//去重判断,若重复则继续-1
end--;
}
}else if(nums[i] + nums[start] + nums[end] < 0){ //小于零则移动start找更大的值
start++;
}else{ //大于零则移动end找更小的值
end--;
}
}
}
}
return result;
}
}
需要注意当找到三个数和为0,记录结果之后,start和end都需要移动,移动的时候也要判断去重的问题。