CodeForces - 776C（前缀和+思维）

链接：CodeForces - 776C

题意：给出数组 a[n] ，问有多少个区间和等于 k^x（x >= 0）。

题解：求前缀和，标记每个和的个数。对每一个数都遍历到1e5，记录到答案。

#include <bits/stdc++.h>
using namespace std;

long long n, k;
map<long long, long long> mp;

int main()
{
    scanf("%lld%lld", &n, &k);

    mp[] = ;
    long long ans = , s = ;
    for(long long i = , a; i < n; i++){
        scanf("%lld", &a);
        s += a;
        if(k == ) ans += mp[s-];
        else if(k == -) ans += mp[s-] + mp[s+];
        else{
            long long t = ;
            while(abs(t) <= 1e15){
                ans += mp[s-t];
                t *= k;
            }
        }
        mp[s]++;
    }

    printf("%lld\n", ans);
    return ;
}