给你一个包含 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]
输出:[]
来源:力扣(LeetCode)
因为求三个数的和,所以直接多个循环使用穷举法一个一个测虽然很简单但是开销太多,所以这里需要先将数据处理一下,第一步先排序,从小到大排行序之后我们会发现如果我们从中选取的三个数第一个大于0那么三数之和必定大于0,根据这一点我们就可以排除很多的组合。第二步挨个遍历大于零的数字(设为第i个),然后定义left和right指针分别指向i+1和nums.length-1。第三步判断三个数字和是否等于零,如果等于0需要加一次判断看left和right相邻的有没有重复值从而避免相同结果,排除后left右移动right左移(这里必须要同时移动,因为在已经排除了相同数字的情况下只移动一个肯定和零不相等),如果三个数大于零,则移动right,让加的最大数变小点,反之移动left
public List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> list=new ArrayList<>();
Arrays.sort(nums);
int left=0;
int right=0;
int sum=0;
for(int i=0;i<nums.length-1;i++){
if(i>0&&nums[i]==nums[i-1])continue;
if(nums[i]<=0){
left=i+1;
right=nums.length-1;
while(left<right){
sum=nums[i]+nums[left]+nums[right];
if(sum==0){
list.add(Arrays.asList(nums[i],nums[left],nums[right]));
while(left<right&&nums[left]==nums[left+1])left++;
while (left<right&&nums[right]==nums[right-1])right--;
left--;
right--;
}
if(sum>0)
right--;
else if(sum<0)
left++;
}
}else
break;
}
return list;
}