标签:while namespace efi amp fine 排序 lld scanf ace
这题用尺取法是怼不出来的。。。
一开始看到区间和等于k的幂,并且有负数,首先想到将前缀和排序后用尺取法,但因为排序后的前缀和次序是乱的,只适用带绝对值的情况(poj2566),所以无法做。
看了题解后发现其实是个简单的dp
//sum[r]=sum[l]+kk; //ans+=mp[sum[i]-kk];其实是个dp #include<bits/stdc++.h> using namespace std; #define ll long long #define maxn 100005 int n,a[maxn]; ll ans,kk,k,sum[maxn]; void solve(ll kk){ map<ll,int>mp; for(int i=0;i<=n;i++){ mp[sum[i]]++; ans+=mp[sum[i]-kk]; } } int main(){ scanf("%d%lld",&n,&k); sum[0]=ans=0; for(int i=1;i<=n;i++){ scanf("%d",&a[i]); sum[i]=sum[i-1]+a[i]; } if(k==1){ solve(1); printf("%lld\n",ans); return 0; } if(k==-1){ solve(1);solve(-1); printf("%lld\n",ans); return 0; } kk=1; while(kk<=(ll)1000000000*n){ solve(kk); kk*=k; } printf("%lld\n",ans); return 0; }
标签:while namespace efi amp fine 排序 lld scanf ace
原文地址:https://www.cnblogs.com/zsben991126/p/10199708.html