Acwing 1230.K倍区间

最新推荐文章于 2024-07-10 16:47:58 发布

skyang.

最新推荐文章于 2024-07-10 16:47:58 发布

阅读量269

点赞数

文章标签：算法

本文链接：https://blog.csdn.net/m0_67286481/article/details/128647559

版权

寒假练习专栏收录该内容

66 篇文章 3 订阅

订阅专栏

题目链接：1230. K倍区间 - AcWing题库

思路：

数列： $a_1 , a_2 , a_3 ... ... a_n$

前缀和： $s_1 , s_2 , s_3 ... ... s_n$

若Si整除k，那么 $a_1$ ~ $a_i$ 是一个k倍区间

若S_i与S_j模k同余，那么 $a_i$ ~ $a_j$ 是一个k倍区间

用数组cnt记录s_i模k的个数

每增加一个ai，增加的k倍区间个数就为：在 $s_1$ ~ $s_{i-1}$ 中，与 $s_i$ 模k同余的元素个数

AC代码：

#include<iostream>
using namespace std;

int a[100005]={0};
int cnt[100005]={1};
int n,k;
long long ans=0;

int main()
{
    cin>>n>>k;
    for(int i=1;i<=n;++i)
    {
        cin>>a[i];
        a[i]=(a[i]+a[i-1])%k;
    }
    
    for(int i=1;i<=n;++i)
    {
        ans+=cnt[a[i]%k];
        cnt[a[i]%k]++;
    }
    
    cout<<ans<<endl;
    
    return 0;
}