776C Molly's Chemicals --- 前缀和

今天又写了一遍，才差不多看懂这道题，不过stl的选择真的好难，靠经验么？

这题的核心就是 sum[r]-sum[l] = pow; 因为r在l的后面 所以要先由r来找l，在放入l，即转化为sum[l]=sum[r]-pow;

#include<bits/stdc++.h>
#include<algorithm>
#include<iostream>
#define mem(a) memset(a,0,sizeof(a))
#define ll long long
#define inf 0x3f3f3f3f
const int N=1e5+5;
const ll MAX=1e14+5;
const int mod=1e9+7;
using namespace std;
int p,q,y[N],tx,ty;
ll a[N];
int n,k,mx;
set<ll>s;
int main()
{
    cin>>n>>k;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
        a[i]=a[i]+a[i-1];
    }
    ll num=1,ans=0;
    for(int i=1;i<=60;i++)
    {
        s.insert(num);
        num*=k;
        if(num>MAX) break;
    }
    set<ll>::iterator it;
    map<ll,ll>mp;
    mp[0]=1; //要考虑从第一个数字开始到某一位的和正好是k的次方的可能
    for(int i=1;i<=n;i++)
    {
        for(it=s.begin();it!=s.end();it++)
        {
            ans+=mp[a[i]-*it];  //这里的a[i]-*it就表示r-pow 即为l
        }
        mp[a[i]]++; //a[i]先要作为r来跟pow来判断之前是否存在l，后来才能作为l
    }
    cout<<ans<<endl;
    return 0;
}