0
点赞
收藏
分享

微信扫一扫

LeetCode18.四数之和

千行 2024-03-14 阅读 14

LeetCode19 四数之和


思路 双指针法

和三数之和思路相似

只不过四数之和需要再多加一层for循环以及更加精细的剪枝操作

首先借助库函数Arrarys.sort(nums)进行快速排序

定义四个指针,i和j分别用于第一层和第二层循环

left和right指针用于定位

如果sum > target right--
如果 sum < targetleft++
如果 sum = targetleft和right同时向中间靠拢一位

剪枝操作(去重)非常容易出错,本人调试断点调试近两个小时(难受)

多试多练是最重要的

import com.sun.net.httpserver.Authenticator.Result;

class Solution {
    public List<List<Integer>> fourSum(int[] nums, int target) {
        Arrays.sort(nums); //快速排序 O(nlogn)
        List<List<Integer>> res = new ArrayList<>();

        int len = nums.length;
        for(int i = 0; i< len - 3; i++){
            //跳出重复
            if(nums[i] > 0 && nums[i] >= target) return res;
            
            //对nums[i]去重
            if(i > 0 && nums[i -1] == nums[i]) continue;

            for (int j = i + 1; j < len - 2; j++){
                if(j > i + 1 && nums[j] == nums[j - 1]){
                    continue;
                }

                int left = j + 1;
                int right = len - 1;
                while(left < right) {
                    int sum = nums[i] + nums[j] + nums[left] + nums[right];
                    if(sum == target){
                        res.add(Arrays.asList(nums[i], nums[j], nums[left], nums[right]));
                    // 对nums[left]去重
                        while(left < right && nums[left+1] == nums[left]) ++left;
                     对nums[right]去重
                        while(left < right && nums[right-1] == nums[right]) --right;
                    //找到答案,双指针同时收缩
                        ++left;
                        --right;
                    }else if( sum > target){
                        --right;
                    }else if(sum < target){
                        ++left;
                    }
                 }
            }
        }
        return res;

    }
}
举报

相关推荐

0 条评论