41.和为S的连续正数序列

题目描述

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

输出描述:

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

思路一：使用DFS，遍历找到所有满足要求的连续序列，因为要求至少有2个数，所有遍历的终止条件是初始位置i<=sum/2

代码一：

class Solution {
public:
    vector<vector<int> > FindContinuousSequence(int sum) {
        vector<vector<int>> res;
        for(int i = 1; i <= sum/2; ++i) {
            vector<int> tmp;
            dfs(res, tmp, sum, 0, i);
        }
        return res;
    }
    
    void dfs(vector<vector<int>>& res, vector<int>& tmp, int sum, int now, int num) {
        if(now == sum) {
            res.push_back(tmp);
        } else if(now > sum) {
            return;
        } else {
            tmp.push_back(num);
            now += num;
            ++num;
            dfs(res, tmp, sum, now, num);
        }
    }
};

思路二：采用双指针，初始时 left=1，right=2，定义当前累加和sum_tmp，若sum_tmp<sum，则++right；若sum_tmp>sum，则--left；并及时更新sum_tmp的值，若sum_tmp=sum，则记录当前结果，并存入res中。

代码二：

class Solution {
public:
    //double point method
    vector<vector<int> > FindContinuousSequence(int sum) {
        vector<vector<int>> res;
        int len = (sum + 1) / 2;
        int left = 1;
        int right = 2;
        int sum_tmp = 3;
        while(left <= len) {
            if(sum_tmp == sum) {
                vector<int> tmp;
                for(int i = left; i <= right; ++i) {
                    tmp.push_back(i);
                }
                res.push_back(tmp);
                sum_tmp -= left;
                ++left;
            } else if(sum_tmp < sum) {
                ++right;
                sum_tmp += right;
            } else {
                sum_tmp -= left;
                ++left;
            }
        }
        return res;
    }
};

小结：对于排好序的数组，在解法上可以优先考虑 双指针解法