3Sum
问题简介:
给定n个整数的数组nums,是否有元素a,b,c在nums中,使a + b + c = 0?
找到数组中所有唯一的三元组,它们的总和为零
注:解决方案集不得包含重复的三元组
例如{1,-1,0}和{1,0,-1}
举例:
给定数组 {-2,1,0,-1,-1,2,3}
输出{
{1,0,-1},
{-1,-1,2},
{-2,0,2},
{-2,-1,3}
}
注:不能{0,0,0}
错误解法:Time Limit Exceeded
三层遍历导致时间复杂度极高
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
Set<List<Integer>> set = new HashSet<>();
int length = nums.length;
if(length < 3) return new ArrayList();
for(int i = 0;i<length-2;i++){
for(int j = i+1;j<length-1;j++){
for(int x = j+1;x<length;x++){
if(nums[i]+nums[j]+nums[x]==0){
Integer [] num = {nums[i],nums[j],nums[x]};
Arrays.sort(num);
List<Integer> list = Arrays.asList(num);
set.add(list);
}
}
}
}
List result = new ArrayList<>(set);
return result;
}
}
正确解法:
可以先将数组排序,做一个判断即排序数组第一个数字大于0或最后一个数字小于0返回空结果
遍历一遍数组,将0与第一个数字做差,差值在剩下的数组里寻找是否有两个数的合值符合要求
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
Arrays.sort(nums);
List<List<Integer>> result = new ArrayList();
int length = nums.length;
if(length ==0 || nums[0]>0 || nums[length-1]<0){
return result;
}
for (int k = 0; k < length; ++k) {
if (nums[k] > 0) break;
if (k > 0 && nums[k] == nums[k - 1]) continue;
int target = 0 - nums[k];
int i = k + 1, j = length - 1;
while (i < j) {
if (nums[i] + nums[j] == target) {
List<Integer> list = new ArrayList();
list.add(nums[k]);
list.add(nums[i]);
list.add(nums[j]);
result.add(list);
while (i < j && nums[i] == nums[i + 1]) ++i;
while (i < j && nums[j] == nums[j - 1]) --j;
++i; --j;
} else if (nums[i] + nums[j] < target) ++i;
else --j;
}
}
return result;
}
}
注:list和set之间转换可以通过初始化
例如已有一个set集合想将其转为list
List list = new ArrayList(set);
小白刷题之路,请多指教— — 要么大器晚成,要么石沉大海