题目描述
小明很喜欢数学,有一天他在做数学作业时,要求计算出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;
}
};
小结:对于排好序的数组,在解法上可以优先考虑
双指针解法