给你一个包含 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]
输出:[]
思路:遍历整个数组,当前元素记为nums[i],随后选取当前元素的右边两端,记为L和R。sum=nums[i]+nums[L]+nums[R]的值,如果sum为0,就添加到ans当中。
判断条件:
(1)如果nums[i]>0,那么其sum的和一定大于0,停止循环。
(2)如果i所指的两个元素,和其前面一个相等,那么就结束本次循环,去重。
要注意的是,新建的list的写法,以及最后添加到list的时候采用的是Arrays.asList()
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> ans = new ArrayList<>();
int len=nums.length;
if(len<3 || nums==null) return ans;
Arrays.sort(nums);
for(int i=0;i<nums.length;i++){
if(nums[i]>0) break;
if(i>0 && nums[i]==nums[i-1]) continue;
int L=i+1;
int R=nums.length-1;
while(L<R){
int sum =nums[i]+nums[L]+nums[R];
if(sum==0){
ans.add(Arrays.asList(nums[i],nums[L],nums[R]));
while(L<R && nums[L]==nums[L+1]) L++;
while(L<R && nums[R]==nums[R-1]) R--;
L++;
R--;
}
else if (sum<0) L++;
else if (sum>0) R--;
}
}
return ans;
}
}