[Leetcode] 18. 四數之和 java
阿新 • • 發佈:2018-12-20
給定一個包含 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; } }