Python蓝桥杯之K倍区间

题目： k倍区间

给定一个长度为N的数列，A1, A2, … AN，如果其中一段连续的子序列Ai, Ai+1, … Aj(i <= j)之和是K的倍数，我们就称这个区间[i, j]是K倍区间。

你能求出数列中总共有多少个K倍区间吗？

输入

第一行包含两个整数N和K。(1 <= N, K <= 100000) 
以下N行每行包含一个整数Ai。(1 <= Ai <= 100000)

输出

输出一个整数，代表K倍区间的数目。

例如， 
输入： 
5 2 
1 
2 
3 
4 
5

程序应该输出： 
6

资源约定： 
峰值内存消耗（含虚拟机） < 256M 
CPU消耗 < 2000ms

请严格按要求输出，不要画蛇添足地打印类似：“请您输入…” 的多余内容。

问题分析：

我们看到输入时要同时输入两个数且要用空格作为分割，所以想到

N, K = map(int, input().split(" "))

代码如下：

N, K = map(int, input().split(" ")) # 同时输入两个数并且用空格隔开
for i in range(1, N+1):
    print(i) # 循环打印i
count = 0 # 先给count赋值为0用来记数，再进入循环
for j in range(1, N+1): 
    sum_num = 0
    for k in range(j, N+1): # 这两个循环就是将所有区间全部列举出来 这里的k是小写
        sum_num += k # sum_num即为区间之和 这里的k为小写
        if sum_num % K == 0: # 判断区间之和与K之间是否有余数 这里的K是大写的
            count += 1 # 符合条件count+1 记数
print(count) # 最后打印记数结果

结果如下图：

 题目做完我突然意识到，不是打印数列而是输入数列

 所有修改代码如下：

N, K = map(int, input().split(" ")) # 同时输入两个数并且用空格隔开
lst = [] # 创建空列表
for i in range(1, N+1):
    f = int(input())
    lst.append(f) # 将输入的f全部存入f之中
count = 0 # 先给count赋值为0用来记数，再进入循环
for j in range(lst[0], N+1): # 循环从lst[0]开始
    sum_num = 0
    for k in range(j, N+1): # 这两个循环就是将所有区间全部列举出来 这里的k是小写
        sum_num += k # sum_num即为区间之和 这里的k为小写
        if sum_num % K == 0: # 判断区间之和与K之间是否有余数 这里的K是大写的
            count += 1 # 符合条件count+1 记数
print(count) # 最后打印记数结果

运行结果入下图：