[数据结构与算法]Leetcode-15 三数之和

题目描述

给你一个包含 n 个整数的数组?nums，判断?nums?中是否存在三个元素 a，b，c ，使得?a + b + c = 0 ？请你找出所有和为 0 且不重复的三元组。

注意：答案中不可以包含重复的三元组。

示例 1：

输入：nums = [-1,0,1,2,-1,-4]
输出：[[-1,-1,2],[-1,0,1]]
示例 2：

输入：nums = []
输出：[]
示例 3：

输入：nums = [0]
输出：[]

提示：

0 <= nums.length <= 3000
-10^5 <= nums[i] <= 10^5

1、避免重复

1）三元组中的三个数必须从小到大，即(a,b,c)中b要大于等于a，c要大于等于b

2）同一重循环中已经添加过的值不要再加入（每一重循环代表三元组中的一个数）

2、双指针

1）当我们需要枚举数组中的两个元素时，如果我们发现随着第一个元素的递增，第二个元素是递减的，那么就可以使用双指针的方法，将枚举的时间复杂度从 O(N^2)减少至 O(N)

2）要始终保证左指针是在右指针之前的

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> result=new ArrayList<List<Integer>>();
        List<Integer> list=new ArrayList<Integer>();
        if(nums==null||nums.length==0)
            return result;
        Arrays.sort(nums);
        for(int i=0;i<nums.length;i++)
        {
            if(i>0&&nums[i]==nums[i-1])
                continue;
            int target=-nums[i];
            int z=nums.length-1;    //初始化z值
            for(int j=i+1;j<nums.length;j++)
            {
                if(j>i+1&&nums[j]==nums[j-1])
                    continue;
                while(j<z&&nums[j]+nums[z]>target)  //z的值是上一次j的遗留值
                    z--;
                if(j==z)
                    break;
                else if(nums[j]+nums[z]==target)
                {
                    list.add(nums[i]);
                    list.add(nums[j]);
                    list.add(nums[z]);
                    result.add(list);
                    list=new ArrayList<Integer>();
                }
            }
        }
        return result;
    }
}