剑指之和为S的连续正数序列

题目描述

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

思路

• 首先我们用low和high分别表示序列的最小值和最大值，把它两分别初始化，最小值为1，最大值为2
• 计算low到high序列的和，可以用等差数列求和公式
• 当序列的和 < s时，增大high；当序列的和 > s时，减少low;当序列的和刚好等于s时，将此序列放入集合中
• 因为序列至少要有两个和，所以 low < high，且low < (1 + sum) /2，可以减少无用的比较次数

import java.util.ArrayList;
public class Solution {
    ArrayList<ArrayList<Integer>> result = new ArrayList<>();
    public ArrayList<ArrayList<Integer> > FindContinuousSequence(int sum) {
        int low = 1, high = 2;
        int mid = (1 + sum)  / 2;
        while (low < mid && low < high) {
            //等差数列的求和公式
            int curSum = (low + high) * (high - low + 1) / 2;
            //相等，那么就将窗口范围的所有数添加进结果集
            if (curSum == sum) {
                sequence(low, high);
                high++;
            } else if (curSum < sum) {
                //如果当前序列的值之和小于sum，增大high
                high++;
            } else {
                //如果当前序列的值之和大于sum，去掉较小的值
                low++;
            }
        }
        return result;
    }
    
    public void sequence(int low, int high) {
        ArrayList<Integer> list = new ArrayList<>();
        for (int i = low; i <= high; i++) {
            list.add(i);
        }
        result.add(list);
    }
}