P8649 [蓝桥杯 2017 省 B] k 倍区间

最新推荐文章于 2024-10-28 22:34:40 发布

MGNisme

最新推荐文章于 2024-10-28 22:34:40 发布

阅读量1k

点赞数 14

文章标签：蓝桥杯算法 c++

本文链接：https://blog.csdn.net/m0_73523403/article/details/136994476

版权

文章讲述了如何使用前缀和和模运算解决一个经典问题，即找出给定整数数组中满足特定条件（前缀和对k取模相等）的区间数量。通过统计每个余数出现的次数并应用组合公式计算答案。

摘要由CSDN通过智能技术生成

P8649 [蓝桥杯 2017 省 B] k 倍区间

题目链接

思路

这题很经典，首先用前缀和维护，若区间 $[l, r]$ 符合条件，则 $s_{l-1}\equiv s_r(mod k)$ ，故对于前缀和的余数，若某个余数出现了 $x$ 次，则可在这 $x$ 个坐标中任选两个作为边界即可，即为 $C_{x}^{2}$ ，若余数为零，则单个数也符合，最后在加上去

代码

#include <iostream>
#include <map>
#define int long long
using namespace std;

map <int, int> mp; //记录每个余数出现个数的数组

signed main()
{
	int n, k, ans = 0;
	cin >> n >> k;
	int cnt = 0;
	for(int i = 1;i <= n;i++)
	{
		int x;
		cin >> x;
		ans += (x % k); //计算前缀和
		mp[ans % k]++; //前缀和模 k
		ans %= k;
	}
	for(int i = 0;i < k;i++) cnt += (mp[i] * (mp[i] - 1)) / 2; //根据上述公式计算答案
	cout << cnt + mp[0] << endl; //将单个的加上去
	return 0;
}