[陣列]Leetcode15三數之和

菜菜Pro發表於2020-11-21

Leetcode15
一個包含 n 個整數的陣列 nums,判斷 nums 中是否存在三個元素 a,b,c ,使得 a + b + c = 0 ?請你找出所有滿足條件且不重複的三元組。
注意:答案中不可以包含重複的三元組。
示例:
給定陣列 nums = [-1, 0, 1, 2, -1, -4],
滿足要求的三元組集合為:
[[-1, 0, 1],[-1, -1, 2]]

class Solution {
    List<List<Integer>> res=new ArrayList<>();
    public List<List<Integer>> threeSum(int[] nums) {
        int len=nums.length;
        Arrays.sort(nums);
        if(len<3) return res;
        for(int i=0;i<len;i++){
            if(i>0&&nums[i]==nums[i-1]){
                continue;
            }
            helper(nums,i,0-nums[i]);
        }
        return res;
    }

    public void helper(int[] nums,int start,int target){
        int low=start+1,hight=nums.length-1;
        while(low<hight){
            if(low>start+1&&nums[low]==nums[low-1]){
                low++;
                continue;
            }
            if(nums[low]+nums[hight]==target){
                List<Integer> list=new ArrayList<>();
                list.add(nums[start]);
                list.add(nums[low]);
                list.add(nums[hight]);
                res.add(list);
                low++;
            }else if(nums[low]+nums[hight]>target){
                hight--;
            }else{
                low++;
            }
        }
    }
}

相關文章