其他分享
首页 > 其他分享> > 2022春训第三场

2022春训第三场

作者:互联网

补题链接

[D]

题意:

给出\(k\)和一个序列,求有多少个子区间的和是\(k\)的倍数。 $ 1 \leq n \leq 10^5, 1 \leq k \leq 10^9 $ 。

分析:

签到题,直接用\(map\)记录前面元素的前缀和就行了,或者枚举\(k\)的幂次然后双指针。
但是没过。要注意\(k=1\)的情况,还要注意一直累加\(k\)的幂次来比较的话可能会爆\(long\ long\),要用 $ __int128 $ 。

代码如下
#include<bits/stdc++.h>
using namespace std;
#define ll long long
int const N=1e5+5;
int n,k;
ll a[N];
map<ll,int>mp;
int main()
{
    int T; scanf("%d",&T);
    while(T--)
    {
        mp.clear(); mp[0]++;
        ll ans=0;
        scanf("%d%d",&n,&k);
        for(int i=1;i<=n;i++)
        {
            scanf("%lld",&a[i]); a[i]+=a[i-1];
            mp[a[i]]++;

            if(k==1) ans+=mp[a[i]-1];
            else
            {
                __int128 nw=k;
                while(nw<=a[i]) ans+=mp[a[i]-nw], nw*=k;//爆longlong
            }
        }
        printf("%lld\n",ans);
    }
    return 0;
}

标签:春训,int,ll,long,leq,mp,2022,scanf,第三场
来源: https://www.cnblogs.com/Zinn/p/16068634.html