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欸
题解
前置知识:前缀和,前缀和的标记优化(可以看看洛谷的非常男女),二分查找
- 首先这题跟区间和有关,也不用修改,前缀和没得跑了
- 前缀和如果不标记那得用双层循环搜索,复杂度+++++
- 我们确定了利用flag标记的方法,那么标记什么呢?可以注意到某组数如果对k取余都一样,那么他们的差一定是k的倍数
- 到这里应付一般的标记题已经可以了,但是这题有个神奇的地方,要求是k的非负整数倍。这个条件对前缀和来说就是后面的前缀和不能小于前面的前缀和。
- 那么我们用一个字典来存对k取余的键,值为对应前缀和组成的字典,那么新的值在列表里的index就是前面比它小的个数,也是因为它多产生的答案数(列表里每个≤它的都能与它组成答案)
- 有顺序的列表那当然可以用二分查找来加速啦
那么可以开始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)