目錄
- 題目概述
- 基本思路
- 代碼實作
- 方法一
- 方法二
- 總結
題目概述
給你一個由 n 個整陣列成的陣列 nums ，和一個目標值 target ，請你找出并回傳滿足下述全部條件且不重復的四元組 [nums[a], nums[b], nums[c], nums[d]] （若兩個四元組元素一一對應，則認為兩個四元組重復）：
0 <= a, b, c, d < n
a、b、c 和 d 互不相同
nums[a] + nums[b] + nums[c] + nums[d] == target
你可以按 任意順序 回傳答案 ，
來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/4sum
示例
輸入：nums = [1,0,-1,0,-2,2], target = 0
輸出：[[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]]
輸入：nums = [2,2,2,2,2], target = 8
輸出：[[2,2,2,2]]
基本思路
使用雙指標法
先將陣列排序，for回圈取到兩個數之后，再建立左指標，右指標，左指標靠近第二層for回圈，右指標放在陣列的尾端，（其實可以理解為四指標法，兩個for回圈也可以算作兩個指標），若sum小于target左指標右移，若sum大于target右指標左移，如此實作指標的動態性，
注：一旦找到符合要求的結果，則兩個指標要同時內縮，否則陷入死回圈！
代碼實作
方法一
public List<List<Integer>> solution(int[] nums,int target){
//建立結果容器(因為要反復添加考慮到重復問題，故使用HashSet結構)
Set<List<Integer>> box = new HashSet<>();
//將陣列排序
Arrays.sort(nums);
//建立左右指標
int left;
int right;
for(int i = 0;i < nums.length;i++){
for(int p = i + 1;p < nums.length;p++){
left = p + 1;
right = nums.length - 1;
while(left < right){
if(nums[i]+nums[p]+nums[left]+nums[right] < target){
left++;
}else if(nums[i]+nums[p]+nums[left]+nums[right] > target){
right--;
}else {
box.add(Arrays.asList(new Integer[]{nums[i],nums[p],nums[left],nums[right]}));
right--;
left++;
}
}
}
}
return new ArrayList<>(box);
}
方法二
public List<List<Integer>> fourSum(int[] nums, int target) {
List<List<Integer>> result = new ArrayList<>();
Arrays.sort(nums);
for (int i = 0; i < nums.length; i++) {
if (i > 0 && nums[i - 1] == nums[i]) {
continue;
}
for (int j = i + 1; j < nums.length; j++) {
if (j > i + 1 && nums[j - 1] == nums[j]) {
continue;
}
int left = j + 1;
int right = nums.length - 1;
while (right > left) {
int sum = nums[i] + nums[j] + nums[left] + nums[right];
if (sum > target) {
right--;
} else if (sum < target) {
left++;
} else {
result.add(Arrays.asList(nums[i], nums[j], nums[left], nums[right]));
while (right > left && nums[right] == nums[right - 1]) right--;
while (right > left && nums[left] == nums[left + 1]) left++;
left++;
right--;
}
}
}
}
return result;
}
總結
方法一比較粗糙，效率較低，而方法二加入了一些剪枝操作，更加高效快捷！
轉載請註明出處，本文鏈接：https://www.uj5u.com/qita/401570.html
標籤：其他
上一篇：XDOJ 5-2 單詞統計