LibreOJ #2551.「JSOI2018」列队 可持久化线段树

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/qq_33229466/article/details/80351267

题意

这里写图片描述
n,m5105,1K,ai106,ai两两不同。

分析

只需考虑每一个数的系数是1还是-1。设区间内的学生按坐标从小到大排序后为a1,a2,...,ak,要求集合的区间为[l,r],不难发现若某个i满足ail+i1,则从ai开始,学生坐标的系数都是1,集合坐标的系数都是-1,然后就做完了。
问题在于如何找到分界点。
不难想到一个二分+主席树求区间k大的O(nlog2n)做法,然而并不能通过。
仔细思考一下会发现其实只要在主席树上二分就好了,具体来说就是按照左儿子中学生数量与集合位置的数量的相对大小来讨论下就好了。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>

typedef long long LL;

const int N=500005;
const int inf=2000000;

int n,m,sz,rt[N];
LL ans;
struct tree{int l,r,s,mx;LL w;}t[N*25];

int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

LL get_sum(int l,int r)
{
    if (l>r) return 0;
    return (LL)(l+r)*(r-l+1)/2;
}

void ins(int &d,int l,int r,int x)
{
    int p=d;d=++sz;t[d]=t[p];t[d].s++;t[d].w+=(LL)x;t[d].mx=std::max(t[d].mx,x);
    if (l==r) return;
    int mid=(l+r)/2;
    if (x<=mid) ins(t[d].l,l,mid,x);
    else ins(t[d].r,mid+1,r,x);
}

void query(int d,int p,int l,int r,int x,int y)
{
    if (t[d].s-t[p].s==0||l==r) return;
    int mid=(l+r)/2,u=t[t[d].l].s-t[t[p].l].s,v=std::max(0,mid-x+1);
    if (y<=mid) u=t[t[d].r].s-t[t[p].r].s,ans+=t[t[d].r].w-t[t[p].r].w-get_sum(y-u+1,y),query(t[d].l,t[p].l,l,mid,x,y-u);
    else if (x>mid) ans+=get_sum(x,x+u-1)-(t[t[d].l].w-t[t[p].l].w),query(t[d].r,t[p].r,mid+1,r,x+u,y);
    else if (u>v) ans+=get_sum(x,x+u-1)-(t[t[d].l].w-t[t[p].l].w),query(t[d].r,t[p].r,mid+1,r,x+u,y);
    else if (u==v)
    {
        if (t[t[d].l].mx!=t[t[p].l].mx&&t[t[d].l].mx==mid) ans+=t[t[d].r].w-t[t[p].r].w-get_sum(mid+1,y),query(t[d].l,t[p].l,l,mid,x,mid);
        else ans+=get_sum(x,mid)-(t[t[d].l].w-t[t[p].l].w),query(t[d].r,t[p].r,mid+1,r,mid+1,y);
    }
    else ans+=t[t[d].r].w-t[t[p].r].w-get_sum(x+u,y),query(t[d].l,t[p].l,l,mid,x,x+u-1);
}

int main()
{
    n=read();m=read();
    for (int i=1,x;i<=n;i++) x=read(),rt[i]=rt[i-1],ins(rt[i],1,inf,x);
    while (m--)
    {
        int l=read(),r=read(),K=read();
        ans=0;
        query(rt[r],rt[l-1],1,inf,K,K+r-l);
        printf("%lld\n",ans);
    }
    return 0;
}
阅读更多
相关热词
换一批

没有更多推荐了,返回首页