快手2020实习生招聘及应届生补录-算法类笔试B卷
快手2020实习生招聘及应届生补录-算法类笔试B卷
第三题:(python),没来得及提交…
import sys for line in sys.stdin: k, n = int(line.split()[0]), int(line.split()[1]) a = [0] * (n + 1) if k == 1: a[n] = 1 print(a[n] % 397) else: # k>= 2 a[0] = 1 a[1] = 1 if n < 2: # print(a[n]) print(a[n] % 397) else: a[2] = 2 for i in range(1, n+1): j = 1 a[i] = 0 while j <= k and i >= j: a[i] += a[i - j] j += 1 # print(a[n]) print(a[n] % 397)