15. 三數之和

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

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

例如, 給定數組 nums = [-1, 0, 1, 2, -1, -4],

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

思路:暴力解(N^3)會超時,先排序(nlogn)轉化成a+b=-c雙指針法

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> list = new ArrayList<>();
        if(nums == null && nums.length ==0) return list;
        sort(nums,0,nums.length-1);
        for(int i =0;i<nums.length;i++){
            if(nums[i]>0)  break;
            int start = i+1; int end = nums.length-1;
            int num = nums[i] *-1;
       
            if(i!=0 && nums[i]==nums[i-1])  continue;
            while(start<end){
                    if(nums[start] +nums[end] >num || (end!=nums.length-1 && nums[end]==nums[end+1])) end--;
                else if(nums[start] +nums[end] <num || (start!=i+1&&nums[start]==nums[start-1])) start++; 
                else {
                    List<Integer> list1 = new ArrayList<>();
                    list1.add(nums[i]);
                    list1.add(nums[start]);
                    list1.add(nums[end]);
                    list.add(list1);
                    start++;
                    end--;
                }
            }
        }
        return list;
    }
    private void sort(int[] arr,int start,int end ){
        if(start >end) return ;
        int i = start;
        int j =end;
        int temp =arr[start];
        while(i<j){
            while(i<j && arr[j]>temp) j--;
            while(i<j && arr[i]<=temp ) i++;
            swap(arr,i,j);
        }
        swap(arr,start,i);
        sort(arr,start,i-1);
        sort(arr,i+1,end);
    }
    private void swap(int arr[] ,int i ,int j){
        int temp =arr[i];
        arr[i] =arr[j];
        arr[j] =temp;
    }
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章