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

题目描述

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

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

AC代码

import java.util.ArrayList;
public class Solution {
    public ArrayList<ArrayList<Integer> > FindContinuousSequence(int sum) {
        ArrayList<ArrayList<Integer>> resList = new ArrayList<>();
        if(sum < 3) return resList;
        int p1 = 1;
        int p2 = 2;
        int curSum = 3;
        while(p2 <= (sum + 1) / 2){
            if(curSum == sum){
                addList(p1, p2, resList);
                curSum = curSum - p1 + p2 + 1;
                p1++;
                p2++;
            }
            else if(curSum < sum){
                p2++;
                curSum += p2;
            }
            else{
                curSum -= p1;
                p1++;
            }
        }
        return resList;
    }
    
    private void addList(int p1, int p2, ArrayList<ArrayList<Integer>> resList){
        ArrayList<Integer> list = new ArrayList<>();
        for(int i = p1;i <= p2;i++) list.add(i);
        resList.add(list);
    }
}