小明很喜欢数学,有一天他在做数学作业时,要求计算出9~16的和,他马上就写出了正确答案是100。但是他并不满足于此,他在想究竟有多少种连续的正数序列的和为100(至少包括两个数)。没多久,他就得到另一组连续正数和为100的序列:18,19,20,21,22。现在把问题交给你,你能不能也很快的找出所有和为S的连续正数序列? Good Luck!
输出所有和为S的连续正数序列。序列内按照从小至大的顺序,序列间按照开始数字从小到大的顺序
示例1
输入:
9
返回值:
[[2,3,4],[4,5]]
思路:使用滑动窗口求解,初始滑动窗口范围为1,2。使用高斯公式求和,若和小则增大窗口,和大则减小窗口,相等则将窗口内的数一起输出。注意判断条件为两数相遇或窗口上界到达所求。
代码:
public ArrayList<ArrayList<Integer> > FindContinuousSequence(int sum) {
int a,b;
ArrayList<ArrayList<Integer>> ans = new ArrayList<>();
a = 1;
b = 2;
while(a != b && b != sum)
{
int c = (a + b) * (b - a + 1) / 2;
if(c > sum)
{
a++;
}
else if(c < sum)
{
b++;
}
else
{
ArrayList<Integer> s = new ArrayList<>();
for(int i = a;i <= b;i++)
{
s.add(i);
}
ans.add(s);
a++;
b++;
}
}
return ans;
}