0
点赞
收藏
分享

微信扫一扫

LeetCode练习15:四数之和

艾晓雪 2022-03-21 阅读 78

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
你可以按 任意顺序 返回答案 。

示例:

输入:nums = [1,0,-1,0,-2,2], target = 0
输出:[[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]]

代码:(排序+双指针)

class Solution {
    public List<List<Integer>> fourSum(int[] nums, int target) {
        List<List<Integer>> result = new ArrayList<List<Integer>>();
        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){
                    if(nums[i] + nums[j] > target - nums[left] - nums[right]){
                        right--;
                    }else if(nums[i] + nums[j] < target - nums[left] - nums[right]){
                        left++;
                    }else{
                        List<Integer> list = new ArrayList<Integer>();
                        list.add(nums[i]);
                        list.add(nums[j]);
                        list.add(nums[left]);
                        list.add(nums[right]);
                        result.add(list);
                        while(left < right && nums[right] == nums[right-1]) right--;
                        while(left < right && nums[left] == nums[left+1])   left++;
                        left++;
                        right--;
                    }
                }
            }
        }
        return result;
    }
}
举报

相关推荐

0 条评论