《剑指office》和为S的连续正数序列



题目描述

小明很喜欢数学,有一天他在做数学作业时,要求计算出9~16的和,他马上就写出了正确答案是100。但是他并不满足于此,他在想究竟有多少种连续的正数序列的和为100(至少包括两个数)。没多久,他就得到另一组连续正数和为100的序列:18,19,20,21,22。现在把问题交给你,你能不能也很快的找出所有和为S的连续正数序列? Good Luck!

思路：

看到这题，我首先想到的就是利用队列求解。建一个队列,从1开始，依次把小于S的数入队，当队列中数值的和为S时，存储下这个队列，继续入队下一个数。当队列中数的和大于S时，开始移除队列中的数，直到队列中数的和小于S。当队列中数的和小于S时，接着入队下一个小于S的数。如此循环操作，直到没有小于S的数了。

代码：

import java.util.*;
public class Solution {
    public ArrayList<ArrayList<Integer> > FindContinuousSequence(int sum) {
        Queue<Integer> queue = new LinkedList<Integer>();
        Queue<Integer> queue1 = new LinkedList<Integer>();
        ArrayList<ArrayList<Integer>> list = new ArrayList<ArrayList<Integer>>();
        int num = 0;
        Integer tempNum = 0;
        //当sum小于3时，直接返回空
        if(sum < 3){
            return list;
        }
        for(int i=1;i<=sum;i++){
            //当sum == num时，保存队列中的正数序列
            if(sum == num){
                //创建临时数组接受和为S的连续序列
                ArrayList<Integer> temp = new ArrayList<Integer>();
                while((tempNum=queue.poll())!=null){
                    temp.add(tempNum);
                    queue1.offer(tempNum);
                }
                while((tempNum=queue1.poll())!=null){
                    queue.offer(tempNum);
                }
                list.add(temp);
                queue.offer(i);
                num = num + i;
            }
            //当num<sum时，入队i
            if(num < sum){
                num = num + i;
                queue.offer(i);
            }
            //当num > sum，出队数据
            if(num > sum){
                while(num > sum){
                    num = num - queue.poll();
                }
            }
        }
        return list;
    }
}