kSum题总结

通用:

1. 除非要返回下标,否则一定要数组排序

2. 固定前面的数,双指针移动确定最后两个数,2Sum一直到kSum都使用这样的套路解题

如果要求kSum小于target:

从小到大排序数组(即Arrays.sort()的默认排序) ,这样外围的循环在nums[i] * k > target的时候就可以结束,直接return;

如果要求kSum大于target:

从大到小排序数组,外围循环在nums[i] * k < target的时候结束;

或依然从小到大排序数组,但从i=2到i=length-1做外围循环,即固定最后一个数,在其前的数组部分移动双指针找两个符合条件的数,这样所有nums[i] < target的都可以直接continue。

注意:

答案是否需要去重。若不同下标就算不同数字则不去重,若相同数值算同一数字则要去重。

/**
 * kSum通用解法:进行k-2层循环,并用双指针方法确定最后两个数
 * kSum()是逐层包上循环的操作
 * twoSum()是双指针确定两个数的方法
 */
class Solution {
    public List<List<Integer>> fourSum(int[] nums, int target) {
        Arrays.sort(nums);
        return kSum(nums, target, 0, 4);
    }

    public List<List<Integer>> kSum(int[] nums, int target, int start, int k) {
        List<List<Integer>> res = new ArrayList<>();
        if (start > nums.length - k || nums[start] * k > target || target > nums[nums.length - 1] * k) return res;
        if (k == 2) return twoSum(nums, target, start);
        for (int i = start; i <= nums.length - k; i++) // 固定这层的一个数
            if (i == start || nums[i - 1] != nums[i]) // 去重
                for (var set : kSum(nums, target - nums[i], i + 1, k - 1)) {
                    res.add(new ArrayList<>(Arrays.asList(nums[i]))); // Arrays.asList()方法得到的是一个不可更改的list,常量list,所以必须再根据它构造一个新的list来添加元素
                    res.get(res.size() - 1).addAll(set);
                }
        return res;
    }

    public List<List<Integer>> twoSum(int[] nums, int target, int start) {
        List<List<Integer>> res = new ArrayList<>();
        int lo = start, hi = nums.length - 1;
        while (lo < hi) {
            int sum = nums[lo] + nums[hi];
            if (sum > target || hi < nums.length - 1 && nums[hi] == nums[hi + 1]) hi--;
            else if (sum < target || lo > start && nums[lo] == nums[lo - 1]) lo++;
            else res.add(Arrays.asList(nums[lo++], nums[hi--]));
        }
        return res;
    }
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值