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

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';
}