K倍区间（数组多个区间求和）

最新推荐文章于 2024-01-19 06:00:00 发布

QLUGCL

最新推荐文章于 2024-01-19 06:00:00 发布

阅读量860

点赞数

分类专栏：蓝桥杯文章标签：算法数据结构

本文链接：https://blog.csdn.net/qq_43813373/article/details/108437959

版权

蓝桥杯专栏收录该内容

9 篇文章 2 订阅

订阅专栏

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

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

输入格式
第一行包含两个整数 N 和 K。

以下 N 行每行包含一个整数 Ai。

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

数据范围
1≤N,K≤100000,
1≤Ai≤100000
输入样例：
5 2
1
2
3
4
5
输出样例：
6

题解

首先需要找连续的区间且是k的倍数，于是双重for循环暴力稳过几个样例，最后肯定卡在TE。
于是采用新办法
a[2]=a[1]+a[2],a[3]=a[2]+a[3]…
这样[l,r]区间的和就是a[r]-a[l-1]了，
那么如果(a[r]-a[l-1])%k==0
证明[l,r]符合题意
延伸一步为a[r]%k=a[l-1]%k
这样的话就可以每次都用res数组存储每一元素模K的值，然后根据对应相同的进行匹配

#include<iostream>
using namespace std;
long long a[1000000];
long long res[1000000];
int main(){
	long long n,k,sum=0,ans=0;
	cin>>n>>k;
	for(long long i=0;i<n;i++){
		cin>>a[i];
		sum+=a[i];
		ans+=res[sum%k];
		// 注意每个0都可以自己构成一个k倍区间。 
		if(sum%k==0)ans++;
		res[sum%k]++;
	}
	cout<<ans<<endl;
	//cout<<ans+res[0]<<endl;
	return 0;
}