CodeForces - 776C Molly's Chemicals（思维）

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

题意：给你n个数，求有多少个区间和是k的幂。

思路：正常思路是枚举每个区间看有多少个是k的幂，但是这是O(n2)。所以换个思路：区间和最大也就1e14，区间和可以用前缀和sum[r]-sum[l]=k^x  =>  sum[r]-k^x=sum[l]。左边枚举只需要O(n*(log2(1e14)) ，而sum[l]用一个map存好r左边的前缀和就行了。

注意：当k=1，-1时，幂会出现重复，需要特判一下。

代码：

#include <bits/stdc++.h>
#define INF 0x3f3f3f3f
#define eps 1e-8
#define fuck(x) cout<<"<"<<x<<">"<<endl
#define fi first
#define se second
#define pb push_back 
#define lson l, m, rt<<1
#define rson m+1, r, rt<<1|1
using namespace std;
typedef long long LL;
typedef unsigned long long ull;
typedef pair<LL, LL> pii;
const double PI = acos(-1.0);
const LL INFLL = 0x3f3f3f3f3f3f3f3fll;
const int maxn = 1e5 + 5;
const int mod = 1e9 + 7;


int n,k;
LL a[maxn];
LL sum[maxn];
map<LL,LL>mp;
int main() {
    scanf ("%d%d",&n,&k);
    sum[0]=0;
    mp.clear();
    LL ans=0;
    for (int i=1;i<=n;i++) scanf ("%lld",&a[i]),sum[i]=sum[i-1]+a[i];
    for (int i=1;i<=n;i++){
        mp[sum[i-1]]++;
        LL tmp=1;
        for (int j=0;j<=47;j++){
            if (tmp>1e15||tmp<-1e15) break;
            LL t=sum[i]-tmp;
            ans+=mp[t];
            tmp*=k;
            if (k==1) break;
            if (k==-1&&tmp==1) break;
        }
    }
    printf ("%lld\n",ans);
    return 0;
}