57-2：和为S的连续正数序列（剑指offer第2版Python）

1、题目描述

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

现在把问题交给你,你能不能也很快的找出所有和为S的连续正数序列? Good Luck!

输出描述：

输出所有和为S的连续正数序列。序列内按照从小至大的顺序，序列间按照开始数字从小到大的顺序。

2、代码详解

设定两个指针，先分别指向数字1和数字2，并设这两个指针为small和big，对small和big求和，如果和大于目标值，则从当前和中删除small值，并把small值加一，如果和小于目标值，则把big值加一，再把新的big值加入和中。如果和等于目标值，就输出small到big的序列，同时把big加一并加入和中，继续之前的操作。

# -*- coding:utf-8 -*-
class Solution:
    def FindContinuousSequence(self, tsum):
        # 滑动窗口法（slide window）
        if tsum < 3:
            return []
        small = 1
        big = 2
        middle = (tsum + 1)//2
        curSum = small + big
        output = []
        while small < middle:  # 序列至少有两个数字
            if curSum == tsum:
                output.append(list(range(small, big+1)))
                # 打破平衡，继续寻找下一组和为s的序列
                big += 1
                curSum += big
            elif curSum > tsum:
                curSum -= small  # 原small值划出窗口
                small += 1
            else:
                big += 1
                curSum += big  # 滑入新的big值
        return output

s = Solution()
print(s.FindContinuousSequence(15))
print(s.FindContinuousSequence(100))

变形1

'''
给出一个正整数N和长度L，找出一段长度大于等于L的连续非负整数，他们的和恰好为N。
答案可能有多个，需要找出长度最小的那个。
例如 N = 18 L = 2：
5 + 6 + 7 = 18
3 + 4 + 5 + 6 = 18
都是满足要求的，但是我们输出更短的 5 6 7
'''
class Solution:
    def findseq(self, N, L):
        if N < 3:
            return []
        small = 1
        big = 2
        mid = (N + 1) // 2
        curN = small + big
        res = []
        while small < mid:
            if curN == N:
                seq = list(range(small, big + 1))
                seqlen = len(seq)
                if seqlen >= L:
                    res.append(seq)
                    big += 1
                    curN += big
            elif curN > N:
                curN -= small
                small += 1
            else:
                big += 1
                curN += big

        return res[-1]
s = Solution()
print(s.findseq(15, 2))
print(s.findseq(100, 5))

相似1

'''
给定一个数组arr，该数组无序，但每个值均为正数，再给定一个正数k。
求arr的所有子数组中所有元素相加和为k的最长子数组长度。
arr = [2,3,5,2,8,4,5,7,8,2,3,4,5,1,9,6]
k = 15
'''
def func(arr, k):
    n = len(arr)
    if n == 0 or k <= 0:
        return
    sum = arr[0]
    left = 0
    right = 0
    max_len = 0

    while right < n:
        if sum == k:
            max_len = max(max_len, right - left + 1)
            sum -= arr[left]
            left += 1
        elif sum < k:
            right += 1
            if right == n:
                break
            sum += arr[right]
        else:
            sum -= arr[left]
            left += 1
    return max_len

arr = [2,3,5,2,8,4,5,7,8,2,3,4,5,1,9,6]  # 2,3,4,5,1
k = 15
# arr = [1, 2, 1, 1, 1]  # 1,1,1
# k = 3
print(func(arr, k))