力扣题目
解题思路
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
解题思路:
这是一个典型的双指针问题。解题步骤如下:
- 对数组进行排序。
- 初始化一个变量
ans
来存储当前的最小差值,并将其设置为Integer的最大值。 - 遍历数组,对于每一个
i
:- 使用双指针
j
和k
,分别指向i+1
和数组的最后一个元素。 - 当
j < k
时,执行以下操作:- 计算三个数
nums[i]
,nums[j]
和nums[k]
的和。 - 如果和等于
0
,说明找到了一个解,将其添加到结果列表中,并结束循环。 - 如果和大于
0
,说明和太大,需要减小和,因此将k
向前移动一位。 - 如果和小于
0
,说明和太小,需要增加和,因此将j
向后移动一位。
- 计算三个数
- 使用双指针
- 返回结果列表。
这个算法的时间复杂度为 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;
}
}