題目描述:
給定一個包含 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]
]
方法:
三指針法:
-
首先判斷nums的長度是否小於4,是則直接返回list
-
將nums排序,此時時間複雜度爲O(nlogn)
-
第一層循環固定k=0,第二層循環固定指針i=k+1,之後再初始化指針m =i+1,j=nums.length-1;
-
判斷sum和target的大小
- 如果(sum==target)將nums[k],nums[i],nums[m],nums[j]加入list,並 j–;m++;
- else if(sum>target) 執行j–;
- else m++;
-
注意隨後要用hashset去重
-
時間複雜度爲:O(nlogn)+O(n^3)=O(n^3)
詳細代碼:
class Solution {
public List<List<Integer>> fourSum(int[] nums, int target) {
List<List<Integer>> list = new ArrayList<>();
if(nums.length<4) return list;
Arrays.sort(nums);
for(int k = 0;k<nums.length;k++){
for(int i =k+1;i<nums.length;i++){
int m =i+1,j=nums.length-1;
while(m<j){
int sum = nums[k]+nums[i]+nums[m]+nums[j];
if(sum==target){
list.add(Arrays.asList(nums[k],nums[i],nums[m],nums[j]));
j--;m++;
}else if(sum>target) j--;
else m++;
}
}
}
HashSet set = new HashSet(list);
list.clear();
list.addAll(set);
return list;
}
}