首页 > 其他 > 详细

k倍区间:前缀和

时间:2019-03-19 10:20:01      阅读:147      评论:0      收藏:0      [点我收藏+]

[蓝桥杯][2017年第八届真题]k倍区间

题目描述

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

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

输入

第一行包含两个整数N和K。(1 <= N, K <= 100000)
以下N行每行包含一个整数Ai。(1 <= Ai <= 100000)

输出

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

样例输入

5 2
1
2
3
4
5

样例输出

6

题目描述:

定理1:n个数,从里边取两个,共有n*(n-1)/2种方法

定理2:前缀和的余数相同,中间区间肯定是k的倍数

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn = 100010;
#define ll long long
ll n, k,t, q[maxn],cnt[maxn],qian=0;		
int main() {
	scanf("%lld%lld", &n, &k);
	for (int i = 1; i <= n; i++) {					//从1开始,方便后边处理
		scanf("%lld", &t);
		qian = (qian + t);							//求输入数组前缀
		q[qian%k]++;								//求前缀和的取余,并用桶排序、统计个数
	}
	q[0]++;											//注意 q[0]=0,所以要自加,否则f[1]会没法判读
	ll flag = 0;
	//for (int i = 0; i <k; i++)
	//	cout << q[i] << " ";
	for (int i = 0; i < k; i++) {
		if(q[i])
		flag += (q[i] * (q[i] - 1) / 2);			//要注意这个公式,n个数,从里边取两个,共有n*(n-1)/2种方法
	}
	cout << flag << "\n";
	return 0;
}

 

k倍区间:前缀和

原文:https://www.cnblogs.com/52dxer/p/10556722.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!