leetcode 15. 三數之和
阿新 • • 發佈:2021-06-13
一、題目
給你一個包含 n 個整數的陣列 nums,判斷 nums 中是否存在三個元素 a,b,c ,使得 a + b + c = 0 ?請你找出所有和為 0 且不重複的三元組。
注意:答案中不可以包含重複的三元組。
二、解法
要使答案不包含重複的三元組,思路是將nums進行排序,並且每次拿出來的元素都不能和上一次的相同。比如[-1,-1,0,1],如果第一次迭代拿出了兩個-1,那麼答案中就會有重複。
對於三重迴圈 \(i,j,k\) ,如果固定了 \(i\) ,隨著 \(nums[j]\) 在迭代中增加,為了滿足 \(nums[i]+nums[j]+nums[k]=0\) 的條件,必然 \(nums[k]\)
class Solution { public List<List<Integer>> threeSum(int[] nums) { Arrays.sort(nums); int len=nums.length; List<List<Integer>> ans=new ArrayList<List<Integer>>(); for(int i=0;i<len;++i){ if(i==0||nums[i]!=nums[i-1]){ //a不能取重複值 int k=len-1; for(int j=i+1;j<len;++j){ if(j==i+1||nums[j]!=nums[j-1]){ //b不能取重複值 while(k>j&&nums[i]+nums[j]+nums[k]>0){ --k; } if(k>j&&nums[i]+nums[j]+nums[k]==0){ List<Integer> cur=new ArrayList<Integer>(); cur.add(nums[i]); cur.add(nums[j]); cur.add(nums[k]); ans.add(cur); } } } } } return ans; } }