题目
题目来源leetcode
leetcode地址:18. 四数之和,难度:中等。
题目描述(摘自leetcode):
给你一个由 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 你可以按 任意顺序 返回答案 。 示例 1: 输入:nums = [1,0,-1,0,-2,2], target = 0 输出:[[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]] 示例 2: 输入:nums = [2,2,2,2,2], target = 8 输出:[[2,2,2,2]] 提示: 1 <= nums.length <= 200 -109 <= nums[i] <= 109 -109 <= target <= 109
本地调试代码:
class Solution { public List<List<Integer>> fourSum(int[] nums, int target) { ... } public static void main(String[] args) { for (List<Integer> integers : new Solution().fourSum(new int[]{1,0,-1,0,-2,2},0)) { System.out.println(Arrays.toString(integers.toArray())); } } }
题解
NO1:双指针
思路: 与三数之和的思路大致相同,只不过多加了一个指针为最后位置(不断向前移动),中间范围中两个指针。我自制了一个动图如下:
代码:
public List<List<Integer>> fourSum(int[] nums, int target) { List<List<Integer>> result = new ArrayList<>(nums.length); Arrays.sort(nums); for (int i = 0; i < nums.length - 3; i++) { //提前剪枝,当前组合数量<4 if(nums.length - 1 - 3 < i){ break; } //与前面一次相同情况直接省略 if(i>0 && nums[i] == nums[i-1]){ continue; } if(nums[i]+nums[i+1]+nums[i+2]+nums[i+3] > target){ break; } //后指针向前次数(与i相同区间) for (int j = nums.length-1; j >= i+3; j--) { if(j < nums.length-1 && nums[j] == nums[j+1]){ continue; } int left = i + 1; int right = j - 1; while(left<right){ int sum = nums[i]+nums[left]+nums[right]+nums[j]; if(sum > target){ right--; }else if(sum < target){ left++; }else{ result.add(Arrays.asList(nums[i],nums[left],nums[right],nums[j])); while(left<right && nums[left]==nums[left+1]){left++;} while(left<right && nums[right]==nums[right-1]){right--;} left++; right--; } } } } return result; }