1. 程式人生 > >JZOJ 5933. 【NOIP2018模擬10.27】百鴿籠

JZOJ 5933. 【NOIP2018模擬10.27】百鴿籠

題目

鴿鴿鴿
在這裡插入圖片描述
有一個序列,有3種操作。
①刪除最左邊的元素
②在序列的最左邊加入一個數。
③查詢序列中第l個數到第r個數中第k小的數。

題解

想到之前ifk通過動態樹最終狀態的dfs序將問題轉化成靜態樹問題,我打算也用類似的方法。
其實不需要。
每次刪除和插入的都是最左邊的。難道不能將序列顛倒一下嗎?
模板題。

心得

主席樹不需要刪除元素,只需要選擇正確的rt[i],在rt[i]的基礎上加元素即可。
PS:將相同大小的陣列放在程式碼的同一行!!!不然很容易開小陣列
每次開陣列的時候留個心眼

程式碼

#include<iostream>
#include
<cstdio>
#include<cstring> #include<algorithm> #include<cmath> #define N 200010 #define M 400010 #define P(a) putchar(a) #define fo(i,a,b) for(i=a;i<=b;i++) #define fd(i,a,b) for(i=a;i>=b;i--) using namespace std; struct note{ int ls,rs,sum; }tr[M*20]; struct noq{ int l,r,k,op;
}qu[N]; int gs,rt[M]; int i,j,k,l,n,m,opx,d2,top,ans; int nl,nr; int o[M],a[M],t2,cx; int c[N]; int read(){ int fh=0,rs=0;char ch=0; while((ch<'0')||(ch>'9')&&(ch^'-'))ch=getchar(); if(ch=='-')fh=1,ch=getchar(); while(ch>='0'&&ch<='9')rs=(rs<<3)+(rs<<1)+(ch^'0'
),ch=getchar(); return fh?-rs:rs; } void write(int x){ if(x>9)write(x/10); P(x%10+'0'); } void change(int &px,int py,int l,int r){ px=++gs; tr[px]=tr[py]; tr[px].sum++; if(l==r)return; int wz=(l+r)>>1; if(opx<=wz)change(tr[px].ls,tr[py].ls,l,wz); else change(tr[px].rs,tr[py].rs,wz+1,r); } int query(int px,int py,int l,int r,int k){ if(l==r)return a[l]; int wz=(l+r)>>1,x=tr[tr[px].ls].sum-tr[tr[py].ls].sum; if(x<k)return query(tr[px].rs,tr[py].rs,wz+1,r,k-x); else return query(tr[px].ls,tr[py].ls,l,wz,k); } int main(){ n=read();m=read(); fo(i,1,n){ c[i]=read(); a[++t2]=c[i]; } fo(i,1,m){ qu[i].op=read(); if(qu[i].op==2){ qu[i].k=read(); a[++t2]=qu[i].k; } if(qu[i].op==3){ qu[i].l=read(),qu[i].r=read(),qu[i].k=read(); } } sort(a+1,a+t2+1); cx=unique(a+1,a+t2+1)-a-1; fo(i,1,n)c[i]=lower_bound(a+1,a+cx+1,c[i])-a; fo(i,1,m)if(qu[i].op==2)qu[i].k=lower_bound(a+1,a+cx+1,qu[i].k)-a; fd(i,n,1){ opx=c[i]; change(rt[i],rt[i+1],1,cx); o[++top]=i; } d2=n; fo(i,1,m){ if(qu[i].op==1){ top--; }else if(qu[i].op==2){ o[++top]=++d2; opx=qu[i].k; change(rt[d2],rt[o[top-1]],1,cx); }else{ nl=o[top-qu[i].r]; nr=o[top-qu[i].l+1]; ans=query(rt[nr],rt[nl],1,cx,qu[i].k); write(ans); P('\n'); } } return 0; }