leetcode-cn 15題： 三數之和

描述

給定一個包含 n 個整數的數組 nums，判斷 nums 中是否存在三個元素 a，b，c ，使得 a + b + c = 0 ？找出所有滿足條件且不重複的三元組。

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

示例：

給定數組 nums = [-1, 0, 1, 2, -1, -4]，

滿足要求的三元組集合爲：
[
[-1, 0, 1],
[-1, -1, 2]
]

Java代碼示例

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/*
 * @lc app=leetcode.cn id=15 lang=java
 *
 * [15] 三數之和
 */

// @lc code=start
class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        Arrays.sort(nums);
        List<List<Integer>> result = new ArrayList<List<Integer>>();

        for(int i=0;i<nums.length-2;i++){
            if(i==0|| (i>0&&nums[i]!=nums[i-1])){
                int left = i+1;
                int right = nums.length-1;
                int sum = 0-nums[i];
                while(left<right){
                    if(nums[left]+nums[right] == sum){
                        result.add(Arrays.asList(nums[i],nums[left],nums[right]));
                        while(left<right&&nums[left]==nums[left+1]){
                            left++;
                        }
                        while(left<right &&nums[right]==nums[right-1]){
                            right--;
                        }
                        left++;
                        right--;
                    }else if(nums[left]+nums[right]<sum){
                        
                        while(left<right&&nums[left]==nums[left+1]){
                            left++;
                        }
                        left++;
                    }else{
                        
                        while(left<right &&nums[right]==nums[right-1]){
                            right--;
                        }
                        right--;

                    }
                }
            }

        }
        return result;
    }
}
// @lc code=end