codeforce1042 D. Petya and Array 树状数组

codeforce1042 D. Petya and Array

相同题型，不过逆序数不用前缀和POJ2299 Ultra-QuickSort 树状数组（求逆序数）+离散化

求区间和小于T的段数

假设的前缀和为,对于每个查询,答案为

准题解：权值线段树（区间第k大）？？

#include<bits/stdc++.h>
using namespace std;
const int MAX=2e5+10;
long long hash_[MAX<<1],a[MAX<<1],t;
int c[MAX<<1],n,cnt;
void add(int x,int v)
{
    while(x<=cnt) c[x]+=v,x+=x&(-x);
}
int get(int r)
{
    int sum=0;
    while(r) sum+=c[r],r-=r&(-r);
    return sum;
}
int main()
{
    long long x;
    while(~scanf("%d%lld",&n,&t))
    {
        memset(c,0,sizeof(c));
        memset(a,0,sizeof(a));
        long long ans=0;
        hash_[0]=0;hash_[1]=-t;a[0]=0;
        for(int i=1;i<=n;++i)
        {
            scanf("%lld",&x);
            a[i]+=a[i-1]+x;
            hash_[i*2]=a[i];
            hash_[i*2+1]=a[i]-t;
        }
        sort(hash_,hash_+n*2+2);
        cnt=unique(hash_,hash_+2*n+2)-hash_;
        add(lower_bound(hash_,hash_+cnt,0)-hash_+1,1);
        for(int i=1;i<=n;++i)
        {
            x=lower_bound(hash_,hash_+cnt,a[i]-t)-hash_+1;
            ans+=i-get(x);
            x=lower_bound(hash_,hash_+cnt,a[i])-hash_+1;
            add(x,1);
        }
        printf("%lld\n",ans);
    }
    return 0;
}