HDU 5700 区间交

枚举起点 二分终点 树状数组check

#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<cstdio>
#include<map>
#include<queue>
#include<vector>
using namespace std;

const int maxn=+;
int n,k,m;
long long a[maxn];
long long sum[maxn];
int c[maxn];

struct X
{
    int L,R;
} s[maxn];

bool cmp(const X&a,const X&b)
{
    return a.L<b.L;
}

int lowbit(int x)
{
    return x&(-x);
}

void update(int pos,int val)
{
    while(pos<=n)
    {
        c[pos]=c[pos]+val;
        pos=pos+lowbit(pos);
    }
}

int getsum(int pos)
{
    int res=;
    while(pos>)
    {
        res=res+c[pos];
        pos=pos-lowbit(pos);
    }
    return res;
}

int main()
{
    while(~scanf("%d%d%d",&n,&k,&m))
    {
        memset(c,,sizeof c);
        sum[]=;
        for(int i=; i<=n; i++)
        {
            scanf("%lld",&a[i]);
            sum[i]=sum[i-]+a[i];
        }
        for(int i=; i<=m; i++) scanf("%d%d",&s[i].L,&s[i].R);
        sort(s+,s++m,cmp);

        long long ans=;
        int now=;
        for(int i=; i<=n; i++)
        {
            while(now<=m&&s[now].L==i)
            {
                update(s[now].R,);now++;

                int p=-;
                int l=i,r=n;

                while(l<=r)
                {
                    int mid=(l+r)/;
                    int q=getsum(n)-getsum(mid-);
                    if(q>k) l=mid+;
                    else if(q<k) r=mid-;
                    else p=mid,l=mid+;
                }

                if(p==-) continue;
                ans=max(ans,sum[p]-sum[i-]);

            }
        }
        printf("%lld\n",ans);
    }
    return ;
}