蓝桥杯2017年第八届真题-k倍区间（前缀和，动态规划）

最新推荐文章于 2024-07-20 09:51:47 发布

大一菜鸟自学后端

最新推荐文章于 2024-07-20 09:51:47 发布

阅读量225

点赞数 5

文章标签：蓝桥杯动态规划算法 c++

本文链接：https://blog.csdn.net/2302_81473103/article/details/136813064

版权

文章讲述了如何通过动态规划和前缀和的性质，利用模运算优化两个循环，解决给定数组中满足特定模条件的子数组对问题，计算它们的组合数。作者给出了C(n,2)的计算方法，包括模为0和不为0的情况。

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

通过这道题目，我们清楚如果用两个for循环暴力遍历，肯定会无法通过所有样例，因此我们可以用前缀和以及动态规划来解决，我们可以发现对于前缀和pre[i]，如果pre[i] % k == 0，那么代表着从开头到第i个前缀和可以整除k，如果(pre[i] - pre[j]) % k == 0,表示从j + 1到 i位的前缀和可以整除，因此我们可以用模以k来替代pre，并用哈希表dp[i]表示模以k为i的数量，对于模为0的，其本身也可以被整除，并且从中挑选两个即可，因此是C（dp[0],2） + dp[0],对于其他情况则是C(dp[i],2),并记录在答案中

上代码

#include<iostream>
#include<cstring>
#include<algorithm>
#define int long long 

using namespace std;

const int N = 1e5 + 10;
int pre[N] = {0};//用于存储前缀和对k的模 
int *a;
int dp[N] = {0};//用于存储每个模对应的数量 
int ans = 0;

signed main(void)
{
	int n, k; cin >> n >> k;
	a = new int [n + 10];
	for(int i = 1; i <= n; i++){
		cin >> a[i];
		pre[i] = (pre[i - 1] + a[i]) % k;//存储数据 
		dp[pre[i]]++;//更新dp数组 
	}
	for(int i = 0; i < k; i++){//模从0开始到k - 1结束 
		if(i == 0 && dp[i]) ans += dp[i] * (dp[i] + 1) / 2;//如果模是0，那么其本身以及随意两个模为0之间都可以整除k，因此是C（dp[i],2） + dp[i] 
		else if(i != 0 && dp[i]) ans += dp[i] * (dp[i] - 1) / 2;  //如果模不是0，那么其本身不能被整除，因此为C(dp[i],2) 
	}
	
	cout << ans << endl; 
	
	return 0;
}