玲珑杯Round#20-E 1157 造物主的戒律 动态主席树算法

当初主要就是为了这里学的主席树。
之前博客已经讲了主席树，这里也相当于模板题，看了前面的讲解应该可以直接AC了。

静态主席树讲解

动态主席树讲解

学动态之前先学静态！！！

题目链接：http://www.ifrog.cc/acm/problem/1157

题意

中文题目就不说了，自己看。

题解

这题唯一的变化就是加了一个<=x和>x的限制条件。
其实转换一下也是比较简单的，比如我们求l,r,x,k1,k2。
先求出区间[l,r]中<=x的数有cnt个，>x的数有tmp个（总数-cnt）。判断一下k1和cnt的大小，如果k1<=cnt就直接求区间[l,r]第k1小数即可否则就输出-1。然后判断k2和tmp的大小，如果k2<=tmp就转换成求区间[l,r]第cnt+k2小数否则输出-1即可。
所以最后查询依旧是按区间第k小查询，只是多加了个判断而已。

我的代码把所有的x也加入到序列再离散化，其实不必要（之后队友提醒后才知道）。我之前是想把x加进去后好找它的位置然后计算cnt，但其实就算不加也是可以直接找到的（TAT。。。）

代码

#include <bits/stdc++.h>using namespace std;const int maxn = 8e5+5;const int maxm = 4e5+5;int T[maxn],L[maxn*20],R[maxn*20],sum[maxn*20];int sz[maxm],h[maxn];int tot,num;struct node{    int l,r,x,k1,k2;}Q[maxm];void build(int& rt,int l,int r){    rt = ++tot;    sum[rt] = 0;    if(l==r) return;    int mid = (l+r)>>1;    build(L[rt],l,mid);    build(R[rt],mid+1,r);}void update(int& rt,int pre,int x,int l,int r){    rt = ++tot;    L[rt] = L[pre];    R[rt] = R[pre];    sum[rt] = sum[pre]+1;    if(l==r) return;    int mid = (l+r)>>1;    if(x<=mid) update(L[rt],L[pre],x,l,mid);    else update(R[rt],R[pre],x,mid+1,r);}int query(int s,int e,int l,int r,int k){    if(l==r) return l;    int res = sum[L[e]]-sum[L[s]];    int mid = (l+r)>>1;    if(k<=res) return query(L[s],L[e],l,mid,k);    else query(R[s],R[e],mid+1,r,k-res);}int query_num(int x,int s,int e,int l,int r){    if(r<=x) return sum[e]-sum[s];    else if(x<l) return 0;    else    {        int mid = (l+r)>>1;        return query_num(x,L[s],L[e],l,mid)+query_num(x,R[s],R[e],mid+1,r);    }}int main(){    int n,m;    num=tot=0;    scanf("%d%d",&n,&m);    for(int i=1;i<=n;i++) scanf("%d",sz+i),h[++num]=sz[i];    for(int i=1;i<=m;i++)    {        scanf("%d%d%d%d%d",&Q[i].l,&Q[i].r,&Q[i].x,&Q[i].k1,&Q[i].k2);        h[++num] = Q[i].x;    }    sort(h+1,h+1+num);    int tmp = unique(h+1,h+1+num)-h-1;    num = tmp;    build(T[0],1,num);    for(int i=1;i<=n;i++) update(T[i],T[i-1],lower_bound(h+1,h+1+num,sz[i])-h,1,num);    int ans,cnt;    for(int i=1;i<=m;i++)    {        cnt=query_num(lower_bound(h+1,h+1+num,Q[i].x)-h,T[Q[i].l-1],T[Q[i].r],1,num);        printf("%d ",Q[i].k1<=cnt ? h[query(T[Q[i].l-1],T[Q[i].r],1,num,Q[i].k1)] : -1);        int tmp = sum[T[Q[i].r]]-sum[T[Q[i].l-1]]-cnt;        printf("%d\n",Q[i].k2<=tmp ? h[query(T[Q[i].l-1],T[Q[i].r],1,num,cnt+Q[i].k2)] : -1);    }    return 0;}