和为S的连续正数序列——JZ41

https://www.nowcoder.com/practice/c451a3fd84b64cb19485dad758a55ebe?tpId=13&&tqId=11194&rp=1&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking

 

描述

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

返回值描述：

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

示例1

输入：

9

复制返回值：

[[2,3,4],[4,5]]

 

 

public class Jz41_FindContinuousSequence {

    @Test
    public void test(){
        ArrayList<ArrayList<Integer>> arrayLists = FindContinuousSequence(100);
        for (ArrayList<Integer> arrayList : arrayLists) {
            for (Integer integer : arrayList) {
                System.out.print(integer+",");
            }
            System.out.println();
        }
    }

    public ArrayList<ArrayList<Integer>> FindContinuousSequence(int sum) {
        ArrayList<ArrayList<Integer>> ans = new ArrayList<>();
        if(sum ==0 || sum == 1){
            return ans;
        }
        int curSum = 1;
        int r = sum / 2 + 1;
        int l = 1;
        for (int i = 1; i <= r; ) {
            if (sum == curSum) {
                ArrayList<Integer> tAns = new ArrayList<>();
                for (int j = l; j <= i; j++) {
                    tAns.add(j);
                }
                ans.add(tAns);
                i++;
                curSum += i;
            }else if(sum > curSum){
                i++;
                curSum += i;
            }else{
                curSum -=l;
                l++;
            }
        }
        return ans;
    }
}