题目描述
小明很喜欢数学,有一天他在做数学作业时,要求计算出9~16的和,他马上就写出了正确答案是100。但是他并不满足于此,他在想究竟有多少种连续的正数序列的和为100(至少包括两个数)。没多久,他就得到另一组连续正数和为100的序列:18,19,20,21,22。现在把问题交给你,你能不能也很快的找出所有和为S的连续正数序列?
输出描述:输出所有和为S的连续正数序列。序列内按照从小至大的顺序,序列间按照开始数字从小到大的顺序。
问题分析
起初看到这个问题的时候我的第一个想法是用sum分别除n(n=2-sum/2)得到一个位置,然后分别从前后取值相加,判断是否等于sum。后来发现这种方法十分难以编写不说,而且很难以判断这一序列是否可用,所以放弃了这一方法。
下面介绍了牛客网一种很好的方法滑动窗法:方法的思想就是取一个滑动的“窗口”,从低位指针low=1开始,向高位high=2依次取数值相加,因为要取的正数序列是连续的,则可以用等差数列求和公式(首项加末项的和乘项数除以二)来得到和num=(low+high)*(high-low+1)/2。
然后进行判断,如果num和sum相等(num==sum),则得到了一个可用序列,将其存入结果集中,low++,low++是因为这一段序列的和刚好与sum相等,若low不变则high++或high–都不可能在取到可用的序列,可用的序列一定在后面;此时若num>sum则说明这一段序列多取了元素,则将low++,low++的原因是之前包含low的序列已经探测过了,而含low的当前序列的和num>sum则应向后继续寻找,所以low++;
而当num<sum时说明少取了元素,则high++,理由和上面类似,只有向后多取元素才可以。
代码及解析
import java.util.ArrayList;
public class Solution {
public ArrayList<ArrayList<Integer> > FindContinuousSequence(int sum) {
ArrayList<ArrayList<Integer>> result =new ArrayList();
int low=1;
int high=2;
while(low<high){
int num=(low+high)*(high-low+1)/2;
if(num==sum){
ArrayList<Integer> arrList=new ArrayList();
for(int i=low;i<=high;i++){
arrList.add(i);
}
result.add(arrList);
low++;
}
if(num<sum){
high++;
}
if(num>sum){
low++;
}
}
return result;
}
}