LeetCode15 三数之和

问题描述：给定一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？找出所有满足条件且不重复的三元组。例如nums = [-1, 0, 1, 2, -1, -4]，满足的三元组集合为[[-1, 0, 1], [-1, -1, 2]]。
思路：

• 暴力遍历

直接穷举解空间，找到所有满足的三元组，同时去除重复元组。
时间复杂度：o(n³)
空间复杂度：o(n)

• 排序+双指针

先将数组排序，从头开始遍历数组，每次固定当前位置，设置两指针分别指向当前位置后续数组的首尾位置。若当前元素或左指针与当前位置之和大于0，可直接跳过后续遍历；否则若此时三者之和大于0，前移右指针，若小于0，后移左指针，直至和等于0或左右指针相遇，等于0时加入元组后需继续判断左右指针是否和下一位置相同，若相同则跳过，避免重复解。
时间复杂度：o(n²)
空间复杂度：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;
    }
}