Leetcode 15. 3Sum

题目描述

Given an array nums of n integers, are there elements a, b, c in nums such that a + b + c = 0? Find all unique triplets in the array which gives the sum of zero.

Note:

The solution set must not contain duplicate triplets.

Example:

Given array nums = [-1, 0, 1, 2, -1, -4],

A solution set is:
[
[-1, 0, 1],
[-1, -1, 2]
]

题意

给出一个序列，在序列中找三个不同的数a,b,c使a+b+c=0(a,b,c不同是指一个序列只能使用同一位置的数一次),而且要求找出的组合序列唯一，即每一种组合只能出现一次([1,-1,0]与[-1,1,0]就不可以)

思路

我们可以先确定一个数a,去找其他两个数b和c，满足b+c=-a就是我们要找的序列，光光这还不够，我们还要思考找到重复序列的问题。我们可以先把序列排序，然后选择数a的时候与前面的元素比较是否相等，如果相等则选下一个元素，直到与前一个元素不相等为止，这就保证了a不重复。保证了a不重复后，我们要找到合适的b和c，因为序列是有序的，所以我们可以设立left与right指针，分别指向a位置的下一个位置和最右边，通过left和right指针移动来找到合适的b和c，找到一组合适的b和c后，还需要移动left和right指针来找其他满足条件的b和c，这里也有可能找到重复的b和c，所以我们在找下一个b和c的时候一定要与前一个b和c不相同，判定方法与找a的一致。
这样我们就能找到符合题目要求的所有序列了。

代码如下

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> list = new ArrayList<List<Integer>>();
        if(nums.length<=2) return list;
        Arrays.sort(nums);
        for (int i=0;i<nums.length;i++)
        {
            if(nums[i]>0) break;
            //确定a元素不重复
            if(i!=0&&nums[i]==nums[i-1]) continue;
            //目标值为-a
            int target=-nums[i];
            //左右指针
            int left=i+1,right=nums.length-1;
            while(left<right)
            {
                //符合条件
                if(nums[left]+nums[right]==target) 
                {
                    list.add(Arrays.asList(nums[i],nums[left],nums[right]));
                    //左右指针移动
                    left++;right--;
                    //另左右指针指向的数与前面的数不同
                    while(left<nums.length&&nums[left]==nums[left-1]) left++;
                    while(right>left&&nums[right]==nums[right+1]) right--;
                }
                //值偏小，left左移
                else if(nums[left]+nums[right]<target) left++;
                //值偏大，right右移
                else right--;
            }
        }
        return list;
    }
}

复杂度

时间复杂度为n*log(n)+n^2
在leetcode上比97%用户的运行速度快。