LeetCode-面试题57.和为s的连续正数序列

LeetCode-面试题57.和为s的连续正数序列

输入一个正整数 target ,输出所有和为 target 的连续正整数序列(至少含有两个数)。

序列内的数字由小到大排列,不同序列按照首个数字从小到大排列。

示例 1:

输入:target = 9
输出:[[2,3,4],[4,5]]
示例 2:

输入:target = 15
输出:[[1,2,3,4,5],[4,5,6],[7,8]]

思路

利用滑动窗口的思想,我们基于target这个“数组”维护一个滑动窗口,窗口内的元素值的和与target进行比较。如果小于target,窗口向右扩大,如果大于target,窗口从左边缩小,如果等于target,则将结果添加进结果集。

class Solution {
    public int[][] findContinuousSequence(int target) {

        List<int[]> res=new ArrayList<int[]>();
        //设置滑动窗口
        int i=1;
        int j=1;
        int sum=0;

        //i不会超过target/2
        while(i<=target/2)
        {
            //窗口右移扩大
            if(sum<target)
            {
                sum+=j;
                j++;
            }
            //窗口缩小
            else if(sum>target)
            {
                sum-=i;
                i++;
            }else
            //计算窗口中数值的和
            {
                int[] tem=new int[j-i];
                for(int m=i;m<j;m++)
                {
                    tem[m-i]=m;
                }
                res.add(tem);
                sum-=i;
                i++;
            }
        }
        return res.toArray(new int[res.size()][]);
    }
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章