尺取+枚举,推出公式以后就是一个枚举加尺取
但是这题的尺取不是对一个值尺取,而是在一个区间内,所以固定左边界,尺取右边界即可
#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 ",ans); } }