1. 程式人生 > 其它 >leetcode 15. 三數之和

leetcode 15. 三數之和

一、題目

給你一個包含 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]\)

要減小,所以 \(j,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;
    }
}