题目描述
小明很喜欢数学,有一天他在做数学作业时,要求计算出9~16的和,他马上就写出了正确答案是100。但是他并不满足于此,他在想究竟有多少种连续的正数序列的和为100(至少包括两个数)。没多久,他就得到另一组连续正数和为100的序列:18,19,20,21,22。现在把问题交给你,你能不能也很快的找出所有和为S的连续正数序列? Good Luck!
输出描述:
输出所有和为S的连续正数序列。序列内按照从小至大的顺序,序列间按照开始数字从小到大的顺序
解法1:
假设这个序列的长度为n,S=(1+n)n/2可得 n^2<2S
当n为奇数时,S/n就是中间数的值。
当n为偶数时,余数是除数的一半,即(sum%i)*2==i。
起始值的计算:a1 + an =2S/n;又有an=a1+(n-1)
a1=S/n-(n-1)/2,
an=S/n+(n-1)/2,
讨论取整的情况,
a1=[s/n]-[(n-1)/2],也满足。
1.n是奇数时s/n和(n-1)/2都是整数,公式成立a1=[s/n]-[(n-1)/2],
2.n是偶数时s/n=[s/n]+0.5,(n-1)/2=[(n-1)/2]+0.5公式依然成立a1=[s/n]-[(n-1)/2],
故:代码也可以写成a1=s/n-(n-1)/2,证毕。
而an=S/n+(n-1)/2就不满足了,因为n为偶数时,因为取整,加起来的值会变小。
所以需要写成(s2+(n-1)n)/(2n)
时间复杂度O(logn);
class Solution {
public:
vector<vector<int> > FindContinuousSequence(int sum) {
vector<vector<int> > ans;
for(int i=sqrt(2*sum);i>=2;i--){
if((i&1)&&(sum%i==0)||(sum%i)*2==i){
vector<int> tmp;
for(int j=sum/i-(i-1)/2;j<=(sum*2+(i-1)*i)/(2*i);j++){
tmp.push_back(j);
}
ans.push_back(tmp);
}
}
return ans;
}
};
解法2:
滑动窗口的思路,时间复杂度O(n);
class Solution {
public:
vector<vector<int> > FindContinuousSequence(int sum) {
vector<vector<int> > ans;
vector<int> tmp;
int l=1,r=2;
while(l<r){
int s=(l+r)*(r-l+1)/2;
if(s==sum){
tmp.clear();
for(int j=l;j<=r;j++){
tmp.push_back(j);
}
ans.push_back(tmp);
l++;
}else if(s<sum){
r++;
}else{
l++;
}
}
return ans;
}
};