1. 程式人生 > >[Leetcode] 18. 四數之和 java

[Leetcode] 18. 四數之和 java

 給定一個包含 n 個整數的陣列 nums 和一個目標值 target,判斷 nums 中是否存在四個元素 a,b,c 和 d ,使得 a + b + c + d 的值與 target 相等?找出所有滿足條件且不重複的四元組。

注意:

答案中不可以包含重複的四元組。

示例:

給定陣列 nums = [1, 0, -1, 0, -2, 2],和 target = 0。

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

和15 16題類似。

class Solution {
    public List<List<Integer>> fourSum(int[] nums, int target) {
		List<List<Integer>> result=new ArrayList<>();
        if(nums.length<4) return result;
        Arrays.sort(nums);
        for(int i=0;i<nums.length;i++){
            if(i>0&&nums[i]==nums[i-1]) continue;
            for(int j=i+1;j<nums.length;j++){
                if(j>i+1&&nums[j]==nums[j-1]) continue;
                int left=j+1;
                int right=nums.length-1;                  
                while(left<right&&i<j){
                    int flag=target-(nums[i]+nums[j]);
                    if(nums[left]+nums[right]==flag){
                        List<Integer> list=new ArrayList<>();
                        list.add(nums[i]);
                        list.add(nums[left]);
                        list.add(nums[right]);
                        list.add(nums[j]);
                        result.add(list);
                        while(left<right&&nums[left]==nums[left+1]) left++;
                        while(left<right&&nums[right]==nums[right-1]) right--;
                        left++;
                        right--;
                    }
                    else if(nums[left]+nums[right]>flag) right--;
                    else left++;
                }
            }
        }
        return result;
    }
}