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

题目

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

解题思路

第一种：暴力解法

1、遍历1到n/2+1，从其开始，求它的连续j项和；
2、如果大于sum，则直接求下一个；
3、如果等于sum，则把这些数放入结果集。

代码如下

import java.util.ArrayList;
public class Solution {
    public ArrayList<ArrayList<Integer> > FindContinuousSequence(int sum) {
        ArrayList<ArrayList<Integer>> res = new ArrayList<>();
        for (int i = 1; i <= sum >> 1; i++) {
            for (int j = 2; j < sum; j++) {
                double temp = (i + i + j-1) * 0.5 * j;
                if (temp == sum) {
                    ArrayList<Integer> list = new ArrayList<>();
                    for (int k = i; k <= i+j-1; k++) {
                        list.add(k);
                    }
                    res.add(list);
                    break;
                } else if (temp > sum) {
                    break;
                }
            }
        }
        return res;
    }
}

提交结果

第二种：使用双指针滑动窗口

第一步：使用2个指针low和high；
第二步：求low到high的和temp；
第三步：
如果temp == sum，则把low到high之间的数加起来，并low++；
如果temp > sum，则说明含有low的前n项和已经满足不了连续和为sum的条件，所以low++；
如果temp < sum，则high++。
这个思路有点难想到，是看了牛友的思路之后写的

ArrayList<ArrayList<Integer>> res = new ArrayList<>();
        int low = 1, high = 2;
        while (low < high) {
            double temp = (low + high) * 0.5 * (high - low + 1);
            if (temp == sum) {
                ArrayList<Integer> list = new ArrayList<>();
                for (int i = low; i <= high; i++) {
                    list.add(i);
                }
                res.add(list);
                low++;
            } else if (temp > sum) {
                low++;
            } else {
                high++;
            }
        }
        return res;

提交结果