【题目描述】
小明很喜欢数学,有一天他在做数学作业时,要求计算出9~16的和,他马上就写出了正确答案是100。但是他并不满足于此,他在想究竟有多少种连续的正数序列的和为100(至少包括两个数)。没多久,他就得到另一组连续正数和为100的序列:18,19,20,21,22。现在把问题交给你,你能不能也很快的找出所有和为S的连续正数序列? Good Luck!
返回值描述:
输出所有和为S的连续正数序列。
1.序列内按照从小至大的顺序;
2.序列间按照开始数字从小到大的顺序
示例
输入 | 输出 |
---|---|
9 | [[2, 3, 4], [4, 5]] |
【解法】
1.暴力解法
class Solution {
public:
vector<vector<int> > FindContinuousSequence(int sum) {
vector<vector<int>> ret;
//由于【至少包含两个连续数字】所以,数组的最大值为sum/2
// 左边界
for (int i=1; i<=sum/2; ++i) {
// 右边界
for (int j=i+1; j<sum; ++j) {
//官方解法是j<sum,但是实际上j<=sum/2+1 即可,例如:4+5=9 int 9/2=4
//因为假设数组的长度为l 那么数组的数字就在sum/l左右
int tmp = 0;
// 求区间和
for (int k = i; k<=j; ++k) {
tmp += k;
}
if (sum == tmp) {
vector<int> ans;
for (int k=i; k<=j; ++k) ans.push_back(k);
ret.push_back(ans);
}
else if (tmp > sum) {
break;
}
}
}
return ret;
}
};
//改进一点的代码,暴力计算次数少一些 【提前剪枝】处为更新处
class Solution {
public:
vector<vector<int> > FindContinuousSequence(int sum) {
vector<vector<int>> ret;
int tmp = 0;
for (int i=1; i<=sum/2; ++i) {
for (int j=i; j<sum; ++j) {
tmp += j;
if (sum == tmp) {
vector<int> ans;
for (int k=i; k<=j; ++k) ans.push_back(k);
ret.push_back(ans);
}
else if (tmp > sum) {
// 提前剪枝
tmp = 0;
break;
}
}
}
return ret;
}
};
2.滑动窗口法
class Solution {
public:
vector<vector<int> > FindContinuousSequence(int sum) {
vector<vector<int>> ret;
int l = 1, r = 1;
int tmp = 0;
while (l <= sum / 2) {
if (tmp < sum) {
tmp += r;
++r;
}
else if (tmp > sum) {
tmp -= l;
++l;
}
else {
vector<int> ans;
for (int k=l; k<r; ++k)
ans.push_back(k);
ret.push_back(ans);
tmp -= l;
++l;
}
}
return ret;
}
};
//本人小思路:
1.设置一个段长,也就是输出的连续数字的长度l
2.那么输出的这l歌连续数字一定在sum/l附近,依旧也是滑动窗口的方式,但是窗口长度是固定的。
弊端是,窗口的长度如何定义是个费脑筋的事情,定义到多长也是个很麻烦的事情。
显然官方的这个解法更好,窗口长度也会随之改变。