剑指 Offer 57 - II. 和为s的连续正数序列

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

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

 

示例 1:

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

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

限制:

1 <= target <= 10^5

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/he-wei-sde-lian-xu-zheng-shu-xu-lie-lcof

sliding window+剪枝

class Solution {
    public int[][] findContinuousSequence(int target) {
        List<int[]> list=new ArrayList<>();
        for(int l=1,r=1,sum=0;r<target/2+2;r++){
            sum+=r;
            while(sum>target){
                sum-=l;
                l++;
            }
            if(sum==target){
                int[] window=new int[r-l+1];
                for(int i=0;i<r-l+1;i++){
                    window[i]=l+i;
                }
                list.add(window);
                if(window.length==2)
                    break;
            }
        }
        int[][] res=new int[list.size()][];
        for(int i=0;i<res.length;i++){
            res[i]=list.get(i);
        }
        return res;
    }
}

2 pointer

class Solution(object):
    def findContinuousSequence(self, target):
        """
        :type target: int
        :rtype: List[List[int]]
        """
        res=[]
        l=1
        r=2
        while l<r:
            sum=(l+r)*(r-l+1)/2
            if sum<target:
                r+=1
            elif sum>target:
                l+=1
            else:
                res.append([i for i in range(l,r+1)])
                l+=1
                r+=1
        return res

 

posted @ 2020-10-11 19:31  XXXSANS  阅读(157)  评论(0编辑  收藏  举报