问题描述:给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。例如nums = [-1, 0, 1, 2, -1, -4],满足的三元组集合为[[-1, 0, 1], [-1, -1, 2]]。
思路:
- 暴力遍历
直接穷举解空间,找到所有满足的三元组,同时去除重复元组。
时间复杂度:o(n3)
空间复杂度:o(n)
- 排序+双指针
先将数组排序,从头开始遍历数组,每次固定当前位置,设置两指针分别指向当前位置后续数组的首尾位置。若当前元素或左指针与当前位置之和大于0,可直接跳过后续遍历;否则若此时三者之和大于0,前移右指针,若小于0,后移左指针,直至和等于0或左右指针相遇,等于0时加入元组后需继续判断左右指针是否和下一位置相同,若相同则跳过,避免重复解。
时间复杂度:o(n2)
空间复杂度:o(1)
//java
//暴力法
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
if(nums.length < 3) return new ArrayList();
List<List<Integer>> beforeList = new ArrayList<>();//筛选前元组集合,该元组接收所有满足元组(除[0,0,0])
List<List<Integer>> afterList = new ArrayList<>();//筛选后元组集合
boolean flag = true;
for(int i = 0; i < nums.length; ++i){//暴力穷举
for(int j = i+1; j < nums.length; ++j){
for(int k = j+1; k < nums.length; ++k){
//去重部分,难点在于记录的元组顺序可能不一样,也可以采用先对记录元组数字排序再记入,便可简单通过contains函数判断是否有重复元组
List<Integer> ls = new ArrayList<Integer>();
ls.add(nums[i]);
ls.add(nums[j]);
ls.add(nums[k]);
if(nums[i] == 0 && nums[j] == 0 && nums[k] == 0){//若三元组为[0,0,0]的形式,特殊处理,只记录一次
if(flag){ //是否记录
afterList.add(ls);
flag = false;
}
}else if((nums[i] + nums[j] + nums[k]) == 0){ //若不为[0,0,0]的形式
afterList.add(ls);//先加入
for(List<Integer> l: beforeList){//对前三元组进行遍历,如果当前元组存在重复,剔除
if(l.contains(nums[i]) && l.contains(nums[j]) && l.contains(nums[k])){
afterList.remove(ls);
break;
}
}
beforeList.add(ls);
}
}
}
}
return afterList;
}
}
//排序+双指针
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
if(nums.length < 3) return new ArrayList();
Arrays.sort(nums);//排序
List<List<Integer>> list = new ArrayList<>();
for(int i = 0; i < nums.length - 2; ++i){
if(nums[i] > 0) return list;//若首元素大于0,则说明后续不可能存在=0的三元组
if(i > 0 && nums[i] == nums[i-1]) continue;//去掉重复首元素
int left = i + 1;
int right = nums.length - 1;
while(left < right){
if(nums[i] + nums[left] + nums[right] == 0){
list.add(new ArrayList<Integer>(Arrays.asList(nums[i], nums[left], nums[right])));//加入=0三元组
while(left < right && nums[left] == nums[++left]){}//后移左指针元素至不相等位置
while(left < right && nums[right] == nums[--right]){}//后移右指针元素至不相等位置
}else if(nums[i] + nums[left] + nums[right] > 0)//和偏大右指针左移
--right;
else//和偏小左指针右移
++left;
}
}
return list;
}
}