第八届蓝桥杯省赛试题——K倍区间

标题: 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

资源约定:
峰值内存消耗(含虚拟机) < 256M
CPU消耗  < 2000ms
请严格按要求输出,不要画蛇添足地打印类似:“请您输入...” 的多余内容。
注意:
main函数需要返回0;
只使用ANSI C/ANSI C++ 标准;
不要调用依赖于编译环境或操作系统的特殊函数。
所有依赖的函数必须明确地在源文件中 #include <xxx>
不能通过工程设置而省略常用头文件。

提交程序时,注意选择所期望的语言类型和编译器类型。


解题思想:首先判断数据范围,作为蓝桥杯压轴题,用long long 型是无可厚非的,首先输入数据,在存储数据的时候用另一个数组存放它的前i个值之和模k的数。表示前缀和。把省略的第0个数的0给补上之后,将sum中的余数,进行分类存储。两两一样的可以互相组合成为区间。(这里运用到如果sum[i]%k==sum[j]%k),那么i,j组成的区间,一定符合要求。

分类之后用n*(n-1)/2公式计算出最终的符合要求的个数

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll N=100000+10;
ll sum[N],a[N],c[N];       //sum表示前面的和再取模
int main()
{
    ll n,k;
    scanf("%lld%lld",&n,&k);
    for(ll i=1; i<=n; i++)
    {
        scanf("%lld",&a[i]);
        sum[i]=(sum[i-1]+a[i])%k;
    }
    c[0]=1;
    for(ll i=1; i<=n; i++)
        c[sum[i]]++;          //判断一下余数各位1234的个数。
    ll ans=0;					
    for(ll i=0; i<k; i++)
        ans+=c[i]*(c[i]-1)/2;   
    printf("%lld\n",ans);
    return 0;
}

猜你喜欢

转载自blog.csdn.net/zmc1248234377/article/details/79771943