洛谷 P1533 可怜的狗狗(离线+权值线段树)

来源:互联网 发布:java获取文件共享路径 编辑:程序博客网 时间:2024/05/17 06:33

传送门
题解:
我第一感觉是:这不是主席树裸题吗?
然后:这用不着主席树吧。。。
Then:离线,然后vector暴力平衡树乱搞就好了呀!
打代码…………
然后愉快地交了上去,就只有90,TLE了一个点。
果然暴力平衡树还是过不了300000呀。
然后就搞出来我原来的Splay,贴了上去,信心满满地又交了一次。
60,2WA,2TLE,我就放弃了。
重新看这个题,我发现,我就可以搞权值线段树呀,以我自带的小常数,应该不会T的。
打代码………..
AC!
正题:
这个题可以离散化,还要离线下来,将询问区间按左端点排序,然后维护一段区间,不断移动左右端点,添加区间内的值,删除被排出区间外的值,每次不断移动直到与询问区间重合,统计答案,最后按顺序输出就好了。
(p.s:主席树其实就是好多棵权值线段树,我这个就是主席树的简化版,不过跑的快)
代码:

#include<cstdio>#include<cstring>#include<iostream>#include<cmath>#include<algorithm>#include<cstdlib>#include<vector>#define ll long longusing namespace std;inline int read(){    int x=0;char ch=' ';int f=1;    while(ch!='-'&&(ch<'0'||ch>'9'))ch=getchar();    if(ch=='-')f=-1,ch=getchar();    while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar();    return x*f;}int n,m,size;int a[300001];struct block{    int l,r,k,id;    inline bool operator < (const block& y) const {        return l<y.l;    }}b[50001];int ans[50001];int c[300001];inline int Hash(int x){    return lower_bound(c+1,c+size+1,x)-c;}int sum[300001<<2];void update(int rt,int l,int r,int x,int v){    if(l==r){        sum[rt]+=v;        return;    }    int mid=l+r>>1;    if(x<=mid)update(rt<<1,l,mid,x,v);    else update(rt<<1|1,mid+1,r,x,v);    sum[rt]=sum[rt<<1]+sum[rt<<1|1];}int kth(int rt,int l,int r,int k){    if(l==r)return c[l];    int mid=l+r>>1;    if(k<=sum[rt<<1])return kth(rt<<1,l,mid,k);    else return kth(rt<<1|1,mid+1,r,k-sum[rt<<1]);}int main(){    n=read();m=read();    for(int i=1;i<=n;i++){        a[i]=read();        c[i]=a[i];    }    sort(c+1,c+n+1);    size=unique(c+1,c+n+1)-c-1;    for(int i=1;i<=m;i++){        b[i].l=read();b[i].r=read();b[i].k=read();b[i].id=i;    }    sort(b+1,b+m+1);    int l=1,r=0;    for(int i=1;i<=m;i++){        while(r<b[i].r){            update(1,1,size,Hash(a[++r]),1);        }        while(l<b[i].l){            update(1,1,size,Hash(a[l++]),-1);        }        ans[b[i].id]=kth(1,1,size,b[i].k);    }    for(int i=1;i<=m;i++){        printf("%d\n",ans[i]);    }    return 0;}