面试题41(2) 和为s的连续正数序列

面试题41(2) 和为s的连续正数序列

题目描述：

小明很喜欢数学,有一天他在做数学作业时,要求计算出9~16的和,他马上就写出了正确答案是100。但是他并不满足于此,他在想究竟有多少种连续的正数序列的和为100(至少包括两个数)。没多久,他就得到另一组连续正数和为100的序列:18,19,20,21,22。现在把问题交给你,你能不能也很快的找出所有和为S的连续正数序列? Good Luck!

输出描述:
输出所有和为S的连续正数序列。序列内按照从小至大的顺序，序列间按照开始数字从小到大的顺序

思路：

滑动窗口的思路，用left和right分别表窗口的左边界和右边界。窗口不停向右边滑动，计算窗口中全部元素的和curSum，

• 如果curSum < sum，则right++，扩展窗口右边界，使窗口内包含更多的数字。
• 如果curSum > sum，则left++，缩小窗口左边界，使窗口内包含更少的数据。

由于至少需要两个连续数字，所以只需要遍历到中点即可

比如sum=100，当left >=50，right>=51时, left到right窗口内的元素和，一定大于100. 所以没有必要再遍历left > 50的情况了。

import java.util.ArrayList;
public class Solution {
    public ArrayList<ArrayList<Integer> > FindContinuousSequence(int sum) {
        int left = 1, right = 2;
        int mid = sum >> 1;
        int curSum = left + right;
        
        ArrayList<ArrayList<Integer>> result = new ArrayList<>();
        while (left <= mid) {
            if (curSum < sum) {
                right++;
                curSum += right;
            } else {
                if (curSum == sum) {
                    ArrayList<Integer> temp = genListByRange(left, right);
                    result.add(temp);
                }
                curSum -= left;
                left++;
            }
        }
        return result;
    }
    
    private ArrayList<Integer> genListByRange(int left, int right) {
        ArrayList<Integer> result = new ArrayList<>();
        for (int i = left; i <= right; i++) {
            result.add(i);
        }
        return result;
    }
}