【LeeCode】三数之和

【题目描述】

给你一个整数数组 ​​nums​​​ ，判断是否存在三元组 ​​[nums[i], nums[j], nums[k]]​​​ 满足 ​​i != j​​​、​​i != k​​​ 且 ​​j != k​​​ ，同时还满足 ​​nums[i] + nums[j] + nums[k] == 0​​ 。请

你返回所有和为 ​​0​​ 且不重复的三元组。

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

​​https://leetcode.cn/problems/3sum/​​

【示例】

示例1：

输入：nums = [-1,0,1,2,-1,-4]
输出：[[-1,-1,2],[-1,0,1]]
解释：
nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0 。
nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0 。
nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0 。
不同的三元组是 [-1,0,1] 和 [-1,-1,2] 。
注意，输出的顺序和三元组的顺序并不重要。

示例2：

输入：nums = [0,1,1]
输出：[]
解释：唯一可能的三元组和不为 0 。

【代码】

​​学习参考​​

package com.company;

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

/**
 * 2022-10-28
 * 三数之和
 * 一个整数数组 nums ，判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ，同时还满足 nums[i] + nums[j] + nums[k] == 0 。
 * 请你返回所有和为 0 且不重复的三元组。
 * 注意：答案中不可以包含重复的三元组。
 *
 */
public class Solution {
    public static void main(String[] args) {
        int[] nums = {-1,0,1,2,-1,-4};
        List<List<Integer>> list = new ArrayList<>();
        list = threeSum(nums);
        for (List<Integer> x : list) {
            System.out.println(x.toString());
        }
    }

    /**
     * 1. 返回的是个数组，所以我们可以定义List<List<Integer>>
     * 2. 针对数组类问题，先排序是个好方法
     * 3. num[i] > 0 , 则最后的结果一定大于0 pass
     *    num[i] = num[i - 1] 或者 num[i] = num[i+1] 数字重复, pass
     *    sum == 0， num[L]==num[L+1] 结果重复, 跳过，L++
     *    sum == 0， num[R]==num[R-1] 结果重复, 跳过，R--
     *    要求返回3元组，则num.length<3也必然 pass
     * 4. 定义左右指针来进行计算 （注意去重）
     */
    private static  List<List<Integer>>  threeSum(int[] nums) {

        List<List<Integer>> list = new ArrayList<>();

        int len = nums.length;
        if( nums == null || len < 3) return list;

        // 排序
        Arrays.sort(nums);

        for(int i = 0; i < len; i++){
            // 如果当前值大于0  则结果必然大于0  排序过了
            if(nums[i] > 0)  break;
            // 去重, 重复数字，必然pass
            if(i > 0 && nums[i] == nums[i - 1]) continue;

            int L = i + 1;
            int R = len - 1;
            while (L < R){
                int sum = nums[i] + nums[L] + nums[R];
                if(sum == 0){
                    list.add(Arrays.asList(nums[i], nums[L], nums[R]));
                    while (L < R && nums[L] == nums[L + 1]) L++;  // 去重
                    while (L < R && nums[R] == nums[R - 1]) R--;  // 去重
                    L++;
                    R--;
                }
                else if(sum < 0) L++;
                else if(sum > 0) R--;
            }
        }

        return list;
    }
}