CSP-X2024解题报告（T2)

B. [CSP-X 2024 -- T2] 消灭怪兽

内存限制：256 MiB时间限制：1000 ms输入文件：monster.in输出文件：monster.out

题目类型：传统评测方式：文本比较

提交提交记录返回比赛

题目描述

怪兽入侵了地球！

为了抵抗入侵，人类设计出了按顺序排列好的 n 件武器，其中第 i 件武器的攻击力为ai，可以造成 **ai **的伤害。

武器已经排列好了，因此不能改变顺序。某件武器可以单独攻击，也可以与相邻的武器进行组合攻击。

具体来说，每次你可以把相邻的若干个（可以为 1 个，即不进行组合）连续的武器组合起来进行攻击，则攻击力为这些连续的武器攻击力之和。

来自外星的怪兽拥有无敌护盾，不会受到任何伤害。

但是人类在交战过程中发现怪兽有个致命的弱点：每次当受到 k 或 k 的倍数的伤害时，怪兽的无敌护盾就能被打破。

请你帮助人类求出有多少种组合武器的方案，使得造成的伤害能打破怪兽的无敌护盾

输入格式

输入文件为 monster.in。

第一行两个正整数 n, k 如题所述；

第二行为 n 个正整数，其中第 i 个数 ai 表示第 i 件武器的攻击力。

输出格式

输出文件为 monster.out。

输出总共的方案数

样例

输入样例 #1

5 3
1 2 3 4 5

输出样例#1

7

输入样例 #2

10 11
1 4 8 10 16 19 21 25 30 43

输出样例 #2

7

输入样例 #3

6 2
2 2 2 2 2 2

输出样例 #3

21

【样例说明】

样例 1 解释：

k=3，而区间 [1, 2]，[1, 3]，[1, 5]，[2, 4]，[3, 3]，[3, 5]，[4, 5] 的区间和均为 3 或 3 的倍数，故一共有 7 种方案

数据范围与提示

【数据范围】

20% 的数据，n, k ≤ 100；

40% 的数据，n, k ≤ 10000,1 ≤ ai ≤ k；

另外存在10% 的数据，k = 2；

另外存在10% 的数据，所有的 ai 均相等。

100% 的数据，1≤n≤10^6,2≤k≤10^6,1≤ai​≤10^9。

根据同余方程，求出取余k==x的一共有y个，就能加1+2+……+y-1种。

注意：区域k==0的要加一个。

#include<bits/stdc++.h>
using namespace std;
#define in long long
in b[1000005],ma=0,c[1000005],d[1000006],cnt=0;
in n,m,l,a[1000005],r;

int main(){
	//freopen("monster.in","r",stdin);
	freopen("monster.out","w",stdout);
	
	cin>>n>>m;
	for(in i=1;i<=n;i++){
		cin>>a[i];
		a[i]+=a[i-1];
		a[i]%=m;
		ma+=a[i];
		b[a[i]]++;
	}
	for(int i=1;i<=n;i++){
		if(a[i]==0)cnt++;
	}
	for(int i=0;i<m;i++){
		cnt+=(b[i]-1)*b[i]/2;
	}
	cout<<cnt;
	return 0;
}