题干
给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
想法
题干要点提取一下就:
1.找到三个元素相加为0
2.不能有重复
最暴力的想法当然是全部枚举出来,但是太蠢了肯定通不过。
于是我们考虑这样查找:
先将nums排序
三个位置
i j k
i从0到nums.length遍历
对于每一个i,如果nums[i]+nums[j]+nums[k]=0,那么就找到一个,加进去完事。
然后j往后,k往前(只有两个数里小的变大,大的变小才有可能=0;
如果nums[i]+nums[j]+nums[k]<0
那么还不够大,j++
反之k–
接下来我们考虑重复的问题
如果nums[j]=nums[j+1]
那么j++往后
同理
如果nums[k]=numsk-1]
那么k–往前
这样就去重
下方部分代码有更好的优化,看代码就行,很简单
java代码:
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
if(nums == null){
return null;
}
//如果数组长度小于3,返回一个空集合
if(nums.length < 3){
return new ArrayList<>();
}
//对数组nums进行排序
Arrays.sort(nums);
HashSet<List<Integer>> set = new HashSet<>();
//让i从数组下标为0开始跑
for (int i = 0; i < nums.length; i++) {
//j从i的后一个数开始
int j = i + 1;
//k从数组最后一个数(最大的数)往前跑
int k = nums.length - 1;
while(j < k){
if(nums[i] + nums[j] + nums[k] == 0){
List<Integer> list = new ArrayList<>();
list.add(nums[i]);
list.add(nums[j]);
list.add(nums[k]);
set.add(list);
//当有重复的数字出现时,j往前跑,防止结果有重复
while(j < k && nums[j] == nums[j + 1]){
j++;
}
while(j < k && nums[k] == nums[k - 1]){
k--;
}
j++;
k--;
}else if(nums[i] + nums[j] + nums[k] < 0){
j++;
}else{
k--;
}
}
}
return new ArrayList<>(set);
}
}
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
int n = nums.length;
Arrays.sort(nums);
List<List<Integer>> result = new ArrayList<List<Integer>>();
for (int i = 0; i < n; i++) {
int k = i + 1;
int j = n - 1;
int value = nums[i];
if (value > 0) {
break;
}
while (k < j) {
if (nums[k] + nums[j] + value == 0) {
List<Integer> list = new ArrayList<>();
list.add(value);
list.add(nums[k]);
list.add(nums[j]);
result.add(list);
while (k + 1 < j && nums[k] == nums[k + 1]) {
k++;
}
while (j - 1 > k && nums[j] == nums[j - 1]) {
j--;
}
k++;
j--;
} else if (nums[k] + nums[j] + value < 0) {
k++;
} else {
j--;
}
}
while (i < n-1 && nums[i] == nums[i + 1]) {
i++;
}
}
return result;
}
}
class Solution {
public static List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> ans = new ArrayList();
int len = nums.length;
if(nums == null || len < 3) return ans;//nums为空返回空 不足三个数就返回空
Arrays.sort(nums); // 排序
for (int i = 0; i < len ; i++) {//i是最左边的最小的数
if(nums[i] > 0) break; // 如果当前数字大于0,则三数之和一定大于0,所以结束循环
if(i > 0 && nums[i] == nums[i-1]) continue; // 去重
int L = i+1;
int R = len-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;
}
}
思路代码参考https://blog.csdn.net/qq_39241239/article/details/82387016