从零开始刷Leetcode day010 三数之和(3SUM)

三数之和(3SUM)-java解法

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

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

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.

示例：

给定数组 nums = [-1, 0, 1, 2, -1, -4]，

满足要求的三元组集合为：
[
[-1, 0, 1],
[-1, -1, 2]
]

java解法 ：

代码：

class Solution {
    public List<List<Integer>> threeSum(int[] nums) 	{		List<List<Integer>> res = new ArrayList<>();
        if(nums == null || nums.length <=2){
        	//判空处理
            return res;
        }
        Arrays.sort(nums);
        //先进行排序
        for(int i =0;i<nums.length-2;i++){
        //采用三指针算法，排序后第一个指针从左向右走
            if(nums[i]>0){
            /*
            如果最左边的数都大于零，则后面的数均大于0，没必要继续循环
            */
                break;
            }if(i>0&&nums[i]==nums[i-1]){
            //第一个指针去重
                continue;
            }
            int target = -nums[i];
            //确定剩下两个数的和
            //剩下的两个指针，一个在左，一个在右
            int left = i+1;
            int right = nums.length-1;
            while(left<right){
                if(nums[left]+nums[right] == target){
                    res.add(new ArrayList(Arrays.asList(nums[i],nums[left],nums[right])));
                    left++;
                    right--;
                    //找到一组后，为了去重，左右指针皆继续向前
                    while(left<right&&nums[left] == nums[left-1]){//左右指针去重
                        left++;
                    }
                    while(left<right&&nums[right] == nums[right+1]){
                        right--;
                    }
                }if(nums[left]+nums[right]<target){
                    left++;
                }if(nums[left]+nums[right]>target){
                    right--;
                }
            }
        }return res;
    }
}