面试题57 - II. 和为s的连续正数序列

面试题57 - II. 和为s的连续正数序列

题目描述

输入一个正整数 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

解题思路

我们可以利用滑动窗口解决这个问题

• 当窗口的和小于 target 的时候，窗口的和需要增加，所以要扩大窗口，窗口的右边界向右移动
• 当窗口的和大于 target 的时候，窗口的和需要减少，所以要缩小窗口，窗口的左边界向右移动
• 当窗口的和恰好等于 target 的时候，我们需要记录此时的结果。设此时的窗口为 [i, j)[i,j)，那么我们已经找到了一个 ii 开头的序列，也是唯一一个 ii 开头的序列，接下来需要找 i+1i+1 开头的序列，所以窗口的左边界要向右移动

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-8bUPNAL8-1589022518434)(./img/lcof-57-01.png)]

参考题解

因为是连续的正数序列，所以当sum==target时，左边界可以直接向右移动两位，即i += 2，这样子时间复杂度会稍微小一点点

import java.util.ArrayList;
import java.util.List;

/**
 * @author Bob
 * @desc
 * @date 2020/5/9 16:42
 */
public class Solution {
    public int[][] findContinuousSequence(int target) {
        int i = 1,j=1; // i为滑动窗口的左边界，j为滑动窗口的右边界
        int sum = 0;  // 滑动窗口的值
        List<int[]> res = new ArrayList<>();

        while (i <= target/2) {
            if (sum < target) {
                // 右边界向右移动
                sum += j;
                j++;
            }else if (sum > target) {
                // 左边界向右移动
                sum -= i;
                i++;
            }else {
                int[] arr = new int[j - i];
                for (int k = i; k < j; k++) {
                    arr[k-i] = k;
                }
                res.add(arr);
                // 左边界向右移动
                sum -= i;
                i++;
                /*
                改进：因为是连续的序列，向右滑动两位
                sum = sum - 2*i - 1;
                i=i+2;
                 */
            }
        }

        return res.toArray(new int[res.size()][]);
    }

    public static void main(String[] args) {
        System.out.println(new Solution().findContinuousSequence(9).toString());
    }
}