18. 四数之和(Java排序和双指针)

题目

给你一个由 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

来源：力扣（LeetCode）

链接：https://leetcode-cn.com/problems/4sum

著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

解题思路

用排序和双指针的方法

代码

class Solution {
    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] == nums[i-1] ){
                continue;
            }
            for( int j = i+1; j < nums.length; j++ ){
                int left = j+1;
                int right = nums.length - 1;
                //去重,使这次的数和上一次的数不同
                if( j > i+1 && nums[j] == nums[j-1] ){
                    continue;
                }
                while( right > left ){
                    int sum = nums[i] + nums[j] + nums[right] + nums[left];
                    //如果sum>target,说明此时四数之和大了,因为数组是排序后了,所以需要right下标向左移动,使三数之和变小。
                    if( sum > target ){
                        right--;
                    }
                    //如果sum<target,说明此时四数之和小了,所以需要left下标向右移动,使三数之和变大。
                    else if( sum < target ){
                        left++;
                    }
                    //如果sum=target,则将满足的结果放到集合中
                    else{
                        List<Integer> list = new ArrayList<>();
                        list.add(nums[i]);
                        list.add(nums[j]);
                        list.add(nums[right]);
                        list.add(nums[left]);
                        result.add(list);
                        //去重,使下一次的数和这一次的数不同
                        while( right > left && nums[left] == nums[left+1] ){
                            left++;
                        }
                        while( right > left && nums[right] == nums[right-1] ){
                            right--;
                        }
                        left++;
                        right--;
                    }
                }
            }
        }
        return result;
    }
}