题目描述
小明很喜欢数学,有一天他在做数学作业时,要求计算出9~16的和,他马上就写出了正确答案是100。但是他并不满足于此,他在想究竟有多少种连续的正数序列的和为100(至少包括两个数)。没多久,他就得到另一组连续正数和为100的序列:18,19,20,21,22。现在把问题交给你,你能不能也很快的找出所有和为S的连续正数序列? Good Luck!
输出描述:
输出所有和为S的连续正数序列。序列内按照从小至大的顺序,序列间按照开始数字从小到大的顺序
解题思路
最容易想到的思路是O(n^2)的复杂度遍历起始和末尾。将和为s的存入vector。
class Solution {
public:
vector<vector<int> > FindContinuousSequence(int sum) {
vector<int> subsum;
vector<vector<int> > res;
int tmpsum = 0;
for(int i = 0; i<sum/2+3;i++){
tmpsum+=i;
subsum.push_back(tmpsum);
}
for(int i = 0; i<sum/2+2;i++){
for(int j=i+1; j<sum/2+3;j++){
if(subsum[j] - subsum[i] == sum){
vector<int> tmp;
for(int k = i+1;i+1!=j&&k<=j;k++) tmp.push_back(k);
if(i+1!=j) res.push_back(tmp);
}
}
}
return res;
}
};
还有一种是使用滑动窗口的方案。设定窗口左边界lplow为1,右边界rplow为2.然后使用while循环,循环终止条件是lplow>=rplow.
如果窗口内的总长度等于s,则存入vector,如果窗口内的总长度大于vector,则lplow++,如果窗口内的总长度小于vector,则rplow++。
import java.util.ArrayList;
public class Solution {
public ArrayList<ArrayList<Integer> > FindContinuousSequence(int sum) {
//存放结果
ArrayList<ArrayList<Integer> > result = new ArrayList<>();
//两个起点,相当于动态窗口的两边,根据其窗口内的值的和来确定窗口的位置和大小
int plow = 1,phigh = 2;
while(phigh > plow){
//由于是连续的,差为1的一个序列,那么求和公式是(a0+an)*n/2
int cur = (phigh + plow) * (phigh - plow + 1) / 2;
//相等,那么就将窗口范围的所有数添加进结果集
if(cur == sum){
ArrayList<Integer> list = new ArrayList<>();
for(int i=plow;i<=phigh;i++){
list.add(i);
}
result.add(list);
plow++;
//如果当前窗口内的值之和小于sum,那么右边窗口右移一下
}else if(cur < sum){
phigh++;
}else{
//如果当前窗口内的值之和大于sum,那么左边窗口右移一下
plow++;
}
}
return result;
}
}