hdu5358 推公式+在一個區間內的尺取+列舉法
阿新 • • 發佈:2018-12-29
尺取+列舉,推出公式以後就是一個列舉加尺取
但是這題的尺取不是對一個值尺取,而是在一個區間內,所以固定左邊界,尺取右邊界即可
#include<bits/stdc++.h> #define maxn 100005 #define ll long long using namespace std; int t,n,m,k,x,y,z,l,tot; ll cnt,sum[maxn]; int a[maxn]; ll solve(ll L,ll R){//求區間[l,R)內i+j的和 ll ans=0; int l=1,r=0; for(int i=1;i<=n;i++){if(l<i) l=i; if(r<i-1) r=i-1; while(l<=n && sum[l]-sum[i-1]<L) l++;//sum(i,l)必須>=L; while(r+1<=n && sum[r+1]-sum[i-1]<R) r++;//sum(i,r)必須<R if(sum[r]-sum[i-1]<L) continue; ans+=(ll)(r-l+1)*i + (ll)(l+r)*(r-l+1)/2; }return ans; } int main(){ int t; scanf("%d",&t); while(t--){ scanf("%d",&n); sum[0]=0; for(int i=1;i<=n;i++){ scanf("%d",&a[i]); sum[i]=sum[i-1]+a[i]; } ll ans=0; for(int k=1;k<=34;k++) ans+=(ll)k*(solve(pow(2,k-1),pow(2,k))); ans+=solve(0,1); printf("%lld\n",ans); } }