Leetcode-15 三数之和

力扣题目

解题思路

java代码

力扣题目：

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

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

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

示例 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 。

示例 3：

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

提示：

• 3 <= nums.length <= 3000
• -105 <= nums[i] <= 105

解题思路：

这是一个典型的双指针问题。解题步骤如下：

1. 对数组进行排序。
2. 初始化一个变量 ans 来存储当前的最小差值，并将其设置为Integer的最大值。
3. 遍历数组，对于每一个 i：
   1. 使用双指针 j 和 k，分别指向 i+1 和数组的最后一个元素。
   2. 当 j < k 时，执行以下操作：
      • 计算三个数 nums[i]，nums[j] 和 nums[k] 的和。
      • 如果和等于 0，说明找到了一个解，将其添加到结果列表中，并结束循环。
      • 如果和大于 0，说明和太大，需要减小和，因此将 k 向前移动一位。
      • 如果和小于 0，说明和太小，需要增加和，因此将 j 向后移动一位。
4. 返回结果列表。

这个算法的时间复杂度为 O(n^2)，其中 n 是数组的长度。这是因为我们需要对数组进行排序，这需要 O(n log n) 的时间，然后我们需要遍历数组，并对每个元素使用双指针算法，这需要 O(n) 的时间。因此，总体的时间复杂度为 O(n^2)。

java代码：

package org.example;

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

public class Leetcode15 {
    public static void main(String[] args) {
        int[] nums = {-1,0,1,2,-1,-4};
        System.out.println(new Leetcode15().threeSumClosest(nums));
    }

    public List<Integer> threeSumClosest(int[] nums) {
        int n = nums.length;
        Arrays.sort(nums);
        if (n == 3){
            return Arrays.asList(nums[0], nums[1], nums[2]);
        }
        int ans = nums[0] + nums[1] + nums[2];
        List<Integer> list = new ArrayList<>();
        for (int i = 0;i < nums.length;i++){
            int j = i + 1, k = n - 1;
            while (j < k){
                ans = nums[i] + nums[j] + nums[k];
                if (ans == 0){
                    list.add(nums[i]);
                    list.add(nums[j]);
                    list.add(nums[k]);
                    break;
                } else if (ans > 0){
                    k--;
                } else if (ans < 0){
                    j++;
                }

            }
        }
        return list;
    }
}