Codeforces 776C：Molly's Chemicals

Codeforces 776C：Molly's Chemicals

题目链接：http://codeforces.com/contest/776/problem/C

题目大意：给出$n$个数，可以构造出$n^2$个区间，问有多少个区间和是$k$的幂次的区间.

前缀和+枚举

昨天晚上从枚举区间入手愣是想不出来，实际上只需要换个思路枚举$k^x$，用$map[pre[i]]$存储前缀和$pre[i]$，查询是否有区间和为$k^x$的区间也就是判断$map[pre[i]-k^x]$是否存在了.

/*注意$k=1$或$k=-1$的情况*/

代码如下：

#include <iostream>
#include <map>
#define mabs(x) (x<0?-x:x)
using namespace std;
typedef long long ll;
ll n,k,t,pre,a[100005],ans;
map<ll,int>mp;
map<ll,bool>vis;
void init(){
    vis[t]=1;
    pre=0;
    mp.clear();
    mp[0]++;
}
int main(void){
    cin>>n>>k;
    for(int i=0;i<n;++i)cin>>a[i];
    for(t=1;!vis[t]&&mabs(t)<=100000000000000;t*=k){
        for(int i=0;i<n;++i){
            pre+=a[i];
            mp[pre]++;
            ans+=mp[pre-t];
        }
    }
    cout<<ans;
}