蓝桥《k倍区间》python题解

ps.对错误的数据范围进行了更正，用markdown语法对格式进行了优化，增加了传送门

题目传送门：k倍区间 - 蓝桥云课 (lanqiao.cn)https://www.lanqiao.cn/problems/2387/learning/?contest_id=82 

问题描述

一个整数序列 A=(a1​,a2​,⋯,an​) 的区间和为Si,j​=ai​+ai+1​+⋯+aj​ 。

给定整数序列 A 和一个正整数 k, 请问有多少个区间 [i,j] 满足 1≤i≤j≤n 且 Si,j​ 是 k 非负整数倍。

输入格式

输入的第一行包含两个整数n、k, 用一个空格分隔。

第二行包含 n 个整数a1​,a2​,⋯,an​, 相邻的整数之间用一个空格分隔。

输出格式

输出一行包含一个数表示满足条件的区间数量。

样例输入

7 3
1 -1 0 2 2 2 -30

样例输出

7

样例说明

满足条件的区间有 [1,2],[1,3],[1,6],[2,5],[3,3],[3,6],[4,6][1,2],[1,3],[1,6],[2,5],[3,3],[3,6],[4,6] 。

评测用例规模与约定

对于 40% 的评测用例, 1≤n≤500,1≤k≤10;

对于 60% 的评测用例, 1≤n≤2000;

对于所有评测用例, 1≤n≤100000,1≤k≤10^9,−10^9≤ai​≤10^9 。

运行限制

语言	最大运行时间	最大运行内存
C++	1s	256M
C	1s	256M
Java	2s	256M
Python3	3s	256M

ps.这题怎么时间给这么多，最长的也没跑到0.3s欸

题解

前置知识：前缀和，前缀和的标记优化（可以看看洛谷的非常男女），二分查找

1. 首先这题跟区间和有关，也不用修改，前缀和没得跑了
2. 前缀和如果不标记那得用双层循环搜索，复杂度+++++
3. 我们确定了利用flag标记的方法，那么标记什么呢？可以注意到某组数如果对k取余都一样，那么他们的差一定是k的倍数
4. 到这里应付一般的标记题已经可以了，但是这题有个神奇的地方，要求是k的非负整数倍。这个条件对前缀和来说就是后面的前缀和不能小于前面的前缀和。
5. 那么我们用一个字典来存对k取余的键，值为对应前缀和组成的字典，那么新的值在列表里的index就是前面比它小的个数，也是因为它多产生的答案数（列表里每个≤它的都能与它组成答案）
6. 有顺序的列表那当然可以用二分查找来加速啦

那么可以开始AC啦：

code

def check(x):
    """要在相等的右边"""
    if x >= flag[b][-1]:
        flag[b].append(x)
        return len(flag[b])-1
    i, j = 0, len(flag[b])-1
    while i < j:
        m = i+j >> 1
        if flag[b][m] <= x:
            i = m+1
        else:
            j = m
    flag[b].insert(i, x)
    return i


n, k = map(int, input().split())
a = list(map(int, input().split()))
pre = [0]*(n+1)
flag = {0: [0]}
ans = 0
for i in range(n):
    pre[i+1] = pre[i]+a[i]
    b = pre[i+1] % k
    if b not in flag:
        flag[b] = [pre[i+1]]
    else:
        ans += check(pre[i+1])
print(ans)