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

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

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

 1 #include<bits/stdc++.h>
 2 #include<algorithm>
 3 #include<iostream>
 4 #define mem(a) memset(a,0,sizeof(a))
 5 #define ll long long
 6 #define inf 0x3f3f3f3f
 7 const int N=1e5+5;
 8 const ll MAX=1e14+5;
 9 const int mod=1e9+7;
10 using namespace std;
11 int p,q,y[N],tx,ty;
12 ll a[N];
13 int n,k,mx;
14 set<ll>s;
15 int main()
16 {
17     cin>>n>>k;
18     for(int i=1;i<=n;i++)
19     {
20         cin>>a[i];
21         a[i]=a[i]+a[i-1];
22     }
23     ll num=1,ans=0;
24     for(int i=1;i<=60;i++)
25     {
26         s.insert(num);
27         num*=k;
28         if(num>MAX) break;
29     }
30     set<ll>::iterator it;
31     map<ll,ll>mp;
32     mp[0]=1; //要考虑从第一个数字开始到某一位的和正好是k的次方的可能
33     for(int i=1;i<=n;i++)
34     {
35         for(it=s.begin();it!=s.end();it++)
36         {
37             ans+=mp[a[i]-*it];  //这里的a[i]-*it就表示r-pow 即为l
38         }
39         mp[a[i]]++; //a[i]先要作为r来跟pow来判断之前是否存在l，后来才能作为l
40     }
41     cout<<ans<<endl;
42     return 0;
43 }

fall in love

 

转载于:https://www.cnblogs.com/XXrll/p/11158600.html