K倍区间(前缀和优化)

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

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

输入格式

第一行包含两个整数 NN 和 KK。

以下 NN 行每行包含一个整数 AiAi。

输出格式

输出一个整数,代表 KK 倍区间的数目。

数据范围

1≤N,K≤1000001≤N,K≤100000,
1≤Ai≤1000001≤Ai≤100000

输入样例:

5 2
1
2
3
4
5

输出样例:6

题解:

个人认为最难理解的就是cnt[0]=1(记录余数为0的数的个数)这一步,这么初始化就很细节,通常我们的前缀和s[0]没有意义,但是对于此题来说,优化到最后就是找与s[R]%k余数相同的s[l],然后累加,但是记录余数为i的数组cnt[0]序列,我们发现当余数为0时还有一种情况就是他自己本身也可以整除k,所以我们需要每次加上这个序列的同时也加上它本身情况(+1)

AC代码:

#include<stdio.h>
#include<iostream>
#include<algorithm>
typedef long long ll;
const int N=100010;
using namespace std;
int n,k;
ll s[N],cnt[N];

int main()
{
	scanf("%d%d",&n,&k);
	for(int i=1;i<=n;i++)
	{
		scanf("%lld",&s[i]);
		s[i]+=s[i-1];//处理前缀和 
	}
	ll res=0;
	cnt[0]=1;//看题解 
	for(int i=1;i<=n;i++)
	{
		res+=cnt[s[i]%k];//找与s[R]%k余数相同的s[l] 
		cnt[s[i]%k]++;//至少需要两个起步才可以算开始基数 
	}//但是对于余数为0的话不需要大于2才开始计数,因为它本身可以作为一个序列整除 
	printf("%d\n",res);
	return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值