剑指 Offer II 007. 数组中和为 0 的三个数
给你一个整数数组 nums ,判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ,同时还满足 nums[i] + nums[j] + nums[k] == 0 。请
你返回所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]
解释:
nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0 。
nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0 。
nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0 。
不同的三元组是 [-1,0,1] 和 [-1,-1,2] 。 注意,输出的顺序和三元组的顺序并不重要。
一开始想到的就是暴力解法,但是,当题目给出的数组是【0,0,0,0,0,…0】的时候,先进行三重循环,可以得到n的三次方组答案,然后再进行去重操作,得到的答案是null,但是这个过程就耗费了很多时间。
最好的办法就是不重复。
也就是保证i<j<k。
先对数组进行排序;
在第一次循环中,如果取到了和之前的i一样的值,就跳过,也就是continue;
第二次循环里面,对于j的操作和上面一样,但是要注意,当i和j是同样的值时,这种情况当然也是成立的,应该考虑,只需要加一个判断条件,j>i+1即可,就可以将i,j对应元素值相同的情况考虑进去。
(例如,i和j都是-1的时候,在最终的答案中也是一种情况)
在第二重循环里面还可以进行改进,那就是j和k分别从两个方向取值,因为i+j+k=0,所以j+k=-i,由于已经对数组进行排序,所以,当j+k>-i的时候,随着k–,才有可能满足条件;
但是k不能无限减下去,当j==k的时候如果还没出现答案,那就说明此次遍历无效,就break。
最后,在找到答案的同时新建list集合,先存储一个答案,将其加入到ans中,每出现一个答案就加入一个。
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
int len=nums.length;
Arrays.sort(nums);
List<List<Integer>> ans = new ArrayList<List<Integer>>();
for(int i=0;i<len;i++)
{
if(i>0 && nums[i] == nums[i-1]){
continue;
}
int flag=-nums[i];
int k=len-1;
for(int j=i+1;j<len;j++)
{
if(j>i+1&&nums[j]==nums[j-1])
{
continue;
}
while(nums[j]+nums[k]>flag&&j<k)
{
k--;
}
if(j==k)
{
break;
}
if(nums[j]+nums[k]==flag)
{
List<Integer> res = new ArrayList<Integer>();
res.add(nums[i]);
res.add(nums[j]);
res.add(nums[k]);
ans.add(res);
}
}
}
return ans;
}
}