CodeForces - 776C 思维+普通的前缀和

最新推荐文章于 2025-08-06 11:51:08 发布

原创最新推荐文章于 2025-08-06 11:51:08 发布 · 605 阅读

0 ·

CC 4.0 BY-SA版权

思维专栏收录该内容

21 篇文章

订阅专栏

本文针对 CodeForces-776C 的一道算法题进行了尝试解答，但遇到了未知的错误。文章中展示了具体的代码实现过程，并寻求关于 map 数据结构和循环逻辑的改进意见。

https://vjudge.net/problem/CodeForces-776C

参考，http://www.cnblogs.com/WHLdbk/p/6506534.html
然而我的代码却不知道怎么错了。。
错误代码。：

map<LL,LL>mp;
LL sum[mxn];
LL k_t[1000];
int main(){
    int n,k;
    //freopen("in.txt","r",stdin);
    sf("%d%d",&n,&k);
    rep(i,1,n)cin>>sum[i],sum[i]+=sum[i-1];
    k_t[1]=1;int cnt=2;
    if(k==-1)k_t[2]=-1,cnt++;
    else if(k!=1){
        LL tmp=k;
        while(tmp<1e15){
            k_t[cnt++]=tmp;
            tmp*=k;
        }
    }
    for(int i=1;i<=cnt;++i)mp[k_t[i]]=1;
    LL ans=0;
    for(int i=1;i<=n;++i){
        if(mp[sum[i]])ans+=mp[sum[i]];
        for(int j=1;j<cnt;++j){
            mp[sum[i]+k_t[j]]++;
        }
    }
    cout<<ans<<'\n';
}