給定一個包含?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;}}
?