给定一个包含 n 个整数的数组 nums
,判断 nums
中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4],满足要求的三元组集合为:
[[-1, 0, 1],[-1, -1, 2]
]
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;public class Solution {public static void main(String[] args) {int[] nums = { -1, 0, 1, 2, -1, -4 };List<List<Integer>> threeSum = threeSum(nums);for (List<Integer> list : threeSum) {for (Integer sum : list) {System.out.print(sum + " ");}System.out.println("");}}private static List<List<Integer>> threeSum(int[] nums) {// 双指针思想:先对nums排序,三个数拆成1+2形式,第一个数nums[k]一定小于等于0,后两个数通过设置两个指针 i,j 分别指向数组两端:// 如果两指针所指向数nums[i]和nums[j]的和sub大于-nums[k],j指针前移,找一个更小的数;如果两指针所指向数的和小于target,// i指针后移,找一个更大点的数;如果相同,则返回即可。 List<List<Integer>> res = new ArrayList<>();Arrays.sort(nums); // -4, -1, -1, 0, 1, 2for (int k = 0; k < nums.length; k++) {if (nums[k] > 0 || (k > 0 && nums[k] == nums[k - 1])) {continue; // 最小的数大于0不可能再组成结果0了 或者 返回的第一个数相等去重}int i = k + 1, j = nums.length - 1;while (i < j) {int sub = nums[i] + nums[j];if (sub == -nums[k]) {List<Integer> list = new ArrayList<>();list.add(nums[k]);list.add(nums[i]);list.add(nums[j]);res.add(list); // 添加结果while (i < j && nums[i] == nums[i + 1]) { // 第二个数相等去重i++;}while (i < j && nums[j] == nums[j + 1]) { // 第三个数相等去重j--;}i++;j--;} else if (sub > -nums[k]) {j--; // 三个数的和大于0,第三个数减小点} else {i++; // 三个数的和小于0,第二个数增大点}}}return res;}}