《死亡迴圈》傳承槍解密教學
阿新 • • 發佈:2021-09-20
15. 三數之和
給你一個包含 n
個整數的陣列 nums
,判斷 nums
中是否存在三個元素 a,b,c ,使得 a + b + c = 0 ?請你找出所有和為 0
且不重複的三元組。
注意:答案中不可以包含重複的三元組。
示例 1:
輸入:nums = [-1,0,1,2,-1,-4]
輸出:[[-1,-1,2],[-1,0,1]]
示例 2:
輸入:nums = []
輸出:[]
示例 3:
輸入:nums = [0]
輸出:[]
提示:
0 <= nums.length <= 3000
-105 <= nums[i] <= 105
題解思路:排序 + 雙指標
排序 + 雙指標:
class Solution { public List<List<Integer>> threeSum(int[] nums) { int n = nums.length; Arrays.sort(nums); List<List<Integer>> ans = new ArrayList<List<Integer>>(); // 列舉 a for (int first = 0; first < n; ++first) { // 需要和上一次列舉的數不相同 if (first > 0 && nums[first] == nums[first - 1]) { continue; } // c 對應的指標初始指向陣列的最右端 int third = n - 1; int target = -nums[first]; // 列舉 b for (int second = first + 1; second < n; ++second) { // 需要和上一次列舉的數不相同 if (second > first + 1 && nums[second] == nums[second - 1]) { continue; } // 需要保證 b 的指標在 c 的指標的左側 while (second < third && nums[second] + nums[third] > target) { --third; } // 如果指標重合,隨著 b 後續的增加 // 就不會有滿足 a+b+c=0 並且 b<c 的 c 了,可以退出迴圈 if (second == third) { break; } if (nums[second] + nums[third] == target) { List<Integer> list = new ArrayList<Integer>(); list.add(nums[first]); list.add(nums[second]); list.add(nums[third]); ans.add(list); } } } return ans; } }