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

和为S的连续正数序列

题目描述

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

输出描述:

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

想法

一个序列我们假设以 $x$ 开始，总共有 $n$ 个数。那么这个序列就是： $x,x+1,...,x+n-1$。
然后和为 $S=\frac{(x+x+n-1)\ast n}{2}$ 。
我们假设 $x$ 取最小的 $1$，则 $n$ 最大为 $S$。
令 $n$ 取最小的 $2$，则 $x$ 最大约为 $\sqrt{2\ast S}$。
方便起见之间我们直接将两个最大值都设为 $S$。

代码

class Solution {
public:
    vector<vector<int> > FindContinuousSequence(int sum) {
        vector<vector<int> > ans;
        if (sum==0) return ans;
        for (int x=1; x<=sum; x++) {
            for (int n=2; n<=sum; n++) {
                if ((2*x+n-1)*n == sum*2) {
                    vector<int> tmp;
                    for (int i=x; i<x+n; i++) {
                        tmp.push_back(i);
                    }
                    ans.push_back(tmp);
                }
            }
        }
        return ans;
    }
};