给定一个包含 n 个整数的数组 nums
,判断 nums
中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4], 满足要求的三元组集合为: [ [-1, 0, 1], [-1, -1, 2] ]
class Solution { public List<List<Integer>> threeSum(int[] nums) { List<List<Integer>> list = new ArrayList(); Arrays.sort(nums); int a, b, c; for (int i = 0; i < nums.length - 2; ) { a = nums[i]; int j = i + 1; int k = nums.length - 1; while (j < k) { b = nums[j]; c = nums[k]; if (a + b + c == 0) { List<Integer> item = new ArrayList<>(); item.add(a); item.add(b); item.add(c); list.add(item); j++; k--; while (j < k && nums[j] == nums[j - 1]) { j++; } while (j < k && nums[k + 1] == nums[k]) { k--; } } else if (a + b + c > 0) { k--; while (j < k && nums[k + 1] == nums[k]) { k--; } } else { j++; while (j < k && nums[j] == nums[j - 1]) { j++; } } } ++i; while (i < nums.length - 2 && nums[i] == nums[i - 1]) { i++; } } return list; } }