主席樹/函數式線段樹/可持久化線段樹
什麼是主席樹
可持久化數據結構(Persistent data structure)就是利用函數式編程的思想使其支持詢問歷史版本、同時充分利用它們之間的共同數據來減小時間和空間消耗。ios
所以可持久化線段樹也叫函數式線段樹又叫主席樹。算法
可持久化數據結構
在算法執行的過程當中，會發如今更新一個動態集合時，須要維護其過去的版本。這樣的集合稱爲是可持久的。編程
實現持久集合的一種方法時每當該集合被修改時，就將其整個的複製下來，可是這種方法會下降執行速度並佔用過多的空間。數組
考慮一個持久集合S。數據結構
如圖所示，對集合的每個版本維護一個單獨的根，在修改數據時，只複製樹的一部分。ide
稱之爲可持久化數據結構。函數式編程
可持久化線段樹
令 T 表示一個結點，它的左兒子是 left(T)，右兒子是 right(T)。函數
若 T 的範圍是 [L,R]，那麼 left(T) 的範圍是 [L,mid]，right(T) 的範圍是 [mid+1,R]。優化
單點更新
咱們要修改一個葉子結點的值，而且不能影響舊版本的結構。ui
在從根結點遞歸向下尋找目標結點時，將路徑上通過的結點都複製一份。
找到目標結點後，咱們新建一個新的葉子結點，使它的值爲修改後的版本，並將它的地址返回。
對於一個非葉子結點，它至多隻有一個子結點會被修改，那麼咱們對將要被修改的子結點調用修改函數，那麼就獲得了它修改後的兒子。
在每一步都向上返回當前結點的地址，使父結點可以接收到修改後的子結點。
在這個過程當中，只有對新建的結點的操做，沒有對舊版本的數據進行修改。
區間查詢
從要查詢的版本的根節點開始，像查詢普通的線段樹那樣查詢便可。
延遲標記
...
區間第K小值問題
有n個數，屢次詢問一個區間[L,R]中第k小的值是多少。
查詢[1,n]中的第K小值
咱們先對數據進行離散化，而後按值域創建線段樹，線段樹中維護某個值域中的元素個數。
在線段樹的每一個結點上用cnt記錄這一個值域中的元素個數。
那麼要尋找第K小值，從根結點開始處理，若左兒子中表示的元素個數大於等於K，那麼咱們遞歸的處理左兒子，尋找左兒子中第K小的數；
若左兒子中的元素個數小於K，那麼第K小的數在右兒子中，咱們尋找右兒子中第K-(左兒子中的元素數)小的數。
查詢區間[L,R]中的第K小值
咱們按照從1到n的順序依次將數據插入可持久化的線段樹中，將會獲得n+1個版本的線段樹（包括初始化的版本），將其編號爲0~n。
能夠發現全部版本的線段樹都擁有相同的結構，它們同一個位置上的結點的含義都相同。
考慮第i個版本的線段樹的結點P，P中儲存的值表示[1,i]這個區間中，P結點的值域中所含的元素個數；
假設咱們知道了[1,R]區間中P結點的值域中所含的元素個數，也知道[1,L-1]區間中P結點的值域中所包含的元素個數，顯然用第一個個數減去第二個個數，就能夠獲得[L,R]區間中的元素個數。
所以咱們對於一個查詢[L,R]，同步考慮兩個根root[L-1]與root[R]，用它們同一個位置的結點的差值就表示了區間[L,R]中的元素個數，利用這個性質，從兩個根節點，向左右兒子中遞歸的查找第K小數便可。
POJ 2104 K-th Number (HDU 2665)
注意可持久化數據結構的內存開銷很是大，所以要注意儘量的減小沒必要要的空間開支。
1 const int maxn=100001; 2 struct Node{ 3 int ls,rs; 4 int cnt; 5 }tr[maxn*20]; 6 int cur,rt[maxn]; 7 void init(){ 8 cur=0; 9 } 10 inline void push_up(int o){ 11 tr[o].cnt=tr[tr[o].ls].cnt+tr[tr[o].rs].cnt; 12 } 13 int build(int l,int r){ 14 int k=cur++; 15 if (l==r) { 16 tr[k].cnt=0; 17 return k; 18 } 19 int mid=(l+r)>>1; 20 tr[k].ls=build(l,mid); 21 tr[k].rs=build(mid+1,r); 22 push_up(k); 23 return k; 24 } 25 int update(int o,int l,int r,int pos,int val){ 26 int k=cur++; 27 tr[k]=tr[o]; 28 if (l==pos&&r==pos){ 29 tr[k].cnt+=val; 30 return k; 31 } 32 int mid=(l+r)>>1; 33 if (pos<=mid) tr[k].ls=update(tr[o].ls,l,mid,pos,val); 34 else tr[k].rs=update(tr[o].rs,mid+1,r,pos,val); 35 push_up(k); 36 return k; 37 } 38 int query(int l,int r,int o,int v,int kth){ 39 if (l==r) return l; 40 int mid=(l+r)>>1; 41 int res=tr[tr[v].ls].cnt-tr[tr[o].ls].cnt; 42 if (kth<=res) return query(l,mid,tr[o].ls,tr[v].ls,kth); 43 else return query(mid+1,r,tr[o].rs,tr[v].rs,kth-res); 44 }
常數優化的技巧
一種在常數上減少內存消耗的方法：
插入值時候先不要一次新建到底，能留住就留住，等到須要訪問子節點時候再建下去。
這樣理論內存複雜度依然是O(Nlg^2N)，但由於實際上不少結點在查詢時候根本沒用到，因此內存能少用一些。
動態第K小值
每一棵線段樹是維護每個序列前綴的值在任意區間的個數，若是仍是按照靜態的來作的話，那麼每一次修改都要遍歷O(n)棵樹，時間就是O(2*M*nlogn)->TLE。
考慮到前綴和，咱們經過樹狀數組來優化，即樹狀數組套主席樹，每一個節點都對應一棵主席樹，那麼修改操做就只要修改logn棵樹，O(nlognlogn+Mlognlogn)時間是能夠的，可是直接建樹要nlogn*logn（10^7）會MLE。
咱們發現對於靜態的建樹咱們只要nlogn個節點就能夠了，並且對於修改操做，只是修改M次，每次改變倆個值（減去原先的，加上如今的）也就是說若是把全部初值都插入到樹狀數組裏是不值得的，因此咱們分兩部分來作，全部初值按照靜態來建，內存O(nlogn)，而修改部分保存在樹狀數組中，每次修改logn棵樹，每次插入增長logn個節點O(M*logn*logn+nlogn)。
可用主席樹解決的問題
POJ 2104 K-th Number
入門題，求區間第K小數。
1 #include <iostream> 2 #include <cstdio> 3 #include <cstring> 4 #include <algorithm> 5 using namespace std; 6 const int maxn=100001; 7 struct Node{ 8 int ls,rs; 9 int cnt; 10 }tr[maxn*20]; 11 int cur,rt[maxn]; 12 void init(){ 13 cur=0; 14 } 15 inline void push_up(int o){ 16 tr[o].cnt=tr[tr[o].ls].cnt+tr[tr[o].rs].cnt; 17 } 18 int build(int l,int r){ 19 int k=cur++; 20 if (l==r) { 21 tr[k].cnt=0; 22 return k; 23 } 24 int mid=(l+r)>>1; 25 tr[k].ls=build(l,mid); 26 tr[k].rs=build(mid+1,r); 27 push_up(k); 28 return k; 29 } 30 int update(int o,int l,int r,int pos,int val){ 31 int k=cur++; 32 tr[k]=tr[o]; 33 if (l==pos&&r==pos){ 34 tr[k].cnt+=val; 35 return k; 36 } 37 int mid=(l+r)>>1; 38 if (pos<=mid) tr[k].ls=update(tr[o].ls,l,mid,pos,val); 39 else tr[k].rs=update(tr[o].rs,mid+1,r,pos,val); 40 push_up(k); 41 return k; 42 } 43 int query(int l,int r,int o,int v,int kth){ 44 if (l==r) return l; 45 int mid=(l+r)>>1; 46 int res=tr[tr[v].ls].cnt-tr[tr[o].ls].cnt; 47 if (kth<=res) return query(l,mid,tr[o].ls,tr[v].ls,kth); 48 else return query(mid+1,r,tr[o].rs,tr[v].rs,kth-res); 49 } 50 int b[maxn]; 51 int sortb[maxn]; 52 int main() 53 { 54 int n,m; 55 int T; 56 //scanf("%d",&T); 57 //while (T--){ 58 while (~scanf("%d%d",&n,&m)){ 59 init(); 60 for (int i=1;i<=n;i++){ 61 scanf("%d",&b[i]); 62 sortb[i]=b[i]; 63 } 64 sort(sortb+1,sortb+1+n); 65 int cnt=1; 66 for (int i=2;i<=n;i++){ 67 if (sortb[i]!=sortb[cnt]){ 68 sortb[++cnt]=sortb[i]; 69 } 70 } 71 rt[0]=build(1,cnt); 72 for (int i=1;i<=n;i++){ 73 int p=lower_bound(sortb+1,sortb+cnt+1,b[i])-sortb; 74 rt[i]=update(rt[i-1],1,cnt,p,1); 75 } 76 for (int i=0;i<m;i++){ 77 int a,b,k; 78 scanf("%d%d%d",&a,&b,&k); 79 int idx=query(1,cnt,rt[a-1],rt[b],k); 80 printf("%d\n",sortb[idx]); 81 } 82 } 83 return 0; 84 }
SPOJ 3267 D-query
求區間內不重複的數的個數。
掃描數列創建可持久化線段樹，第i個數若第一次出現，則在線段樹中的位置i加1；若不是第一次出現，將上次出現的位置減1，在本次位置加1。
對於每一個詢問的區間 [L,R]，在第R個版本上的線段樹只有前R個數，在線段樹上查詢位置L，對通過的區間中的和進行累計便可。
1 #include <iostream> 2 #include <cstdio> 3 #include <cstring> 4 #include <algorithm> 5 #include <map> 6 using namespace std; 7 const int maxn=100001; 8 struct Node{ 9 int ls,rs; 10 int cnt; 11 }tr[maxn*20]; 12 int cur,rt[maxn]; 13 void init(){ 14 cur=0; 15 } 16 inline void push_up(int o){ 17 tr[o].cnt=tr[tr[o].ls].cnt+tr[tr[o].rs].cnt; 18 } 19 int build(int l,int r){ 20 int k=cur++; 21 if (l==r) { 22 tr[k].cnt=0; 23 return k; 24 } 25 int mid=(l+r)>>1; 26 tr[k].ls=build(l,mid); 27 tr[k].rs=build(mid+1,r); 28 push_up(k); 29 return k; 30 } 31 int update(int o,int l,int r,int pos,int val){ 32 int k=cur++; 33 tr[k]=tr[o]; 34 if (l==pos&&r==pos){ 35 tr[k].cnt+=val; 36 return k; 37 } 38 int mid=(l+r)>>1; 39 if (pos<=mid) tr[k].ls=update(tr[o].ls,l,mid,pos,val); 40 else tr[k].rs=update(tr[o].rs,mid+1,r,pos,val); 41 push_up(k); 42 return k; 43 } 44 int query(int l,int r,int o,int pos){ 45 if (l==r) return tr[o].cnt; 46 int mid=(l+r)>>1; 47 if (pos<=mid) return tr[tr[o].rs].cnt+query(l,mid,tr[o].ls,pos); 48 else return query(mid+1,r,tr[o].rs,pos); 49 } 50 int b[maxn]; 51 map<int,int> mp; 52 int main() 53 { 54 int n,m; 55 //int T; 56 //scanf("%d",&T); 57 //while (T--){ 58 while (~scanf("%d",&n)){ 59 mp.clear(); 60 init(); 61 for (int i=1;i<=n;i++){ 62 scanf("%d",&b[i]); 63 } 64 rt[0]=build(1,n); 65 for (int i=1;i<=n;i++){ 66 if (mp.find(b[i])==mp.end()){ 67 mp[b[i]]=i; 68 rt[i]=update(rt[i-1],1,n,i,1); 69 } 70 else{ 71 int tmp=update(rt[i-1],1,n,mp[b[i]],-1); 72 rt[i]=update(tmp,1,n,i,1); 73 } 74 mp[b[i]]=i; 75 } 76 scanf("%d",&m); 77 for (int i=0;i<m;i++){ 78 int a,b; 79 scanf("%d%d",&a,&b); 80 int ans=query(1,n,rt[b],a); 81 printf("%d\n",ans); 82 } 83 } 84 return 0; 85 }
-
每一个你不满意的现在，都有一个你没有努力的曾经。