JZ57 和为s的连续正数序列

91 阅读1分钟

leetcode.cn/problems/he…

输入一个正整数 target ,输出所有和为 target 的连续正整数序列(至少含有两个数)。

序列内的数字由小到大排列,不同序列按照首个数字从小到大排列。

 

示例 1:

输入:target = 9 输出:[[2,3,4],[4,5]] 示例 2:

输入:target = 15 输出:[[1,2,3,4,5],[4,5,6],[7,8]]  

限制:

1 <= target <= 10^5

方法二:滑动窗口(双指针)

设连续正整数序列的左边界 i 和右边界 j ,则可构建滑动窗口从左向右滑动。循环中,每轮判断滑动窗口内元素和与目标值 target 的大小关系,若相等则记录结果,若大于 target 则移动左边界 i (以减小窗口内的元素和),若小于 target 则移动右边界 j (以增大窗口内的元素和)。

算法流程:

image.png

复杂度分析

image.png

image.png

代码:

观察本文的算法流程发现,当 s=target 和 s>target 的移动边界操作相同,因此可以合并,代码如下所示。

class Solution {
    public int[][] findContinuousSequence(int target) {
        int i = 1, j = 2, s = 3;
        List<int[]> res = new ArrayList<>();
        while(i < j) {
            if(s == target) {
                int[] ans = new int[j - i + 1];
                for(int k = i; k <= j; k++)
                    ans[k - i] = k;
                res.add(ans);
            }
            if(s >= target) {
                s -= i;
                i++;
            } else {
                j++;
                s += j;
            }
        }
        // toArray(new T[0])能够运行是因为java做了优化,能动态生成对应大小的数组,是官方建议的书写方式,能避免某些并发问题并且效率更高
        return res.toArray(new int[0][]);
    }
}