1. 程式人生 > 實用技巧 >三數之和

三數之和

給你一個包含 n 個整數的陣列nums,判斷nums中是否存在三個元素 a,b,c ,使得a + b + c = 0 ?請你找出所有滿足條件且不重複的三元組。

注意:答案中不可以包含重複的三元組。

示例:

給定陣列 nums = [-1, 0, 1, 2, -1, -4],

滿足要求的三元組集合為:
[
[-1, 0, 1],
[-1, -1, 2]
]

思路

標籤:陣列遍歷
首先對陣列進行排序,排序後固定一個數 nums[i],再使用左右指標指向 nums[i]後面的兩端,數字分別為 nums[L] 和 nums[R],計算三個數的和 sum 判斷是否滿足為 0,滿足則新增進結果集
如果 nums[i]大於 
0,則三數之和必然無法等於 0,結束迴圈 如果 nums[i] == nums[i-1],則說明該數字重複,會導致結果重複,所以應該跳過 當 sum == 0 時,nums[L] == nums[L+1] 則會導致結果重複,應該跳過,L++ 當 sum == 0 時,nums[R] == nums[R−1] 則會導致結果重複,應該跳過,R−−
 1 public List<List<Integer>> threeSum(int[] nums) {
 2         List<List<Integer>> list = new
ArrayList(); 3 if(nums==null||nums.length<3){ 4 return list; 5 } 6 Arrays.sort(nums); 7 for(int i = 0;i<nums.length;i++){ 8 if(nums[i]>0){ 9 break; 10 } 11 if(i>0 && nums[i]==nums[i-1]){
12 continue; 13 } 14 int left = i+1; 15 int right = nums.length-1; 16 while(left<right){ 17 int sum = nums[i]+nums[left]+nums[right]; 18 if(sum==0){ 19 list.add(Arrays.asList(nums[i],nums[left],nums[right])); 20 while(left<right&&nums[left]==nums[left+1]){ 21 left++; 22 } 23 while(left<right&&nums[right]==nums[right-1]){ 24 right--; 25 } 26 left++; 27 right--; 28 }else if(sum<0){ 29 left++; 30 }else if(sum>0){ 31 right--; 32 } 33 34 } 35 36 } 37 return list; 38 }