【Python】美团数据分析师笔试的一些小题目

1 题目描述

给出一个正整数N和长度L，找出一段长度大于等于L的连续非负整数，他们的和恰好为N。答案可能有多个，我我们需要找出长度最小的那个。
例如 N = 18 L = 2：
5 + 6 + 7 = 18
3 + 4 + 5 + 6 = 18
都是满足要求的，但是我们输出更短的 5 6 7

输入描述:

输入数据包括一行： 两个正整数N(1 ≤ N ≤ 1000000000),L(2 ≤ L ≤ 100)

输出描述:

从小到大输出这段连续非负整数，以空格分隔，行末无空格。如果没有这样的序列或者找出的序列长度大于100，则输出No

示例1

输入：18 2
输出：5 6 7

尝试了直勾勾的算法，从数字1到N，项数从L到100，挨个检查一遍，复杂度太高了，（我是智障:)
输出结果的要求，最后一个后面不能有空格，也迷了好久。
（我是智障*2)

def search(N,L) :
    result = ['no']
    length = []
    
    for i in range(1,N,L):
        for j in range(L, min(100,N)):
            arr = [i for i in range(i,i+j)]
            if sum(arr) == N:
                #print(arr)
                length.append(j)
                result.append(arr)
                #print(length,result)
    return result[length.index(min(length))+1]


if __name__ == '__main__':
    [N,L] = list(map(int,input().split()))
    l = len(search(N,L))-1
    for i in range(0,l):
        print(search(N,L)[i], end=' ')
        #百度：如何不换行输出
    print(search(N,L)[l])

后来在评论里面发现一种更简单的算法（才发现原来我没上过高中），
根据等差数列求和公式:
$$S_n=n{a}_1+\frac{n(n-1)}{2}\ast d,$$
这里间隔 $d=1$，推导首项与数列和 N 以及数列项数 $n,(L-1<n<101)$的关系：
$$a_1=\frac{(2N+n-n^2)}{2n}.$$

N, L = map(int, input().split())
ans=['no']
for n in range(L,101):
    a = (2*N + n - n**2)//(2*n) #地板除
    if a > 0:     
        arr = [n for n in range(a,a + n)]
        print(n,a,arr)
        if sum(arr) == N:
            ans = [str(n) for n in range(a,a + n)]
            
            print(ans)
            break 
            #第一次满足要求的n就是最短的，所以立刻跳出循环
            #我真是个小机灵鬼er
print(' '.join(ans))
#这个用法我真的记住了，要求ans是str类型的

啊睡了

题目来自牛客网