POJ2104 ---（区间第K 大）

学习划分树和主席树练得模板题。

题目 ：

就是区间第K 大了。

目前只会划分树和主席树了。

Code：

/* 划分树//#include <bits/stdc++.h>#include <iostream>#include <algorithm>#define lson l,m,rt<<1#define rson m+1,r,rt<<1|1using namespace std;const int maxn = 100000 +69;int tree[30][maxn];         //记录树第dep层序列int sorted[maxn];           //原始序列排序后int toleft[30][maxn];       //记录树第dep层 前 i 个有多少进入左区间。void Build(int l, int r, int dep) {    if(l ==r) return ;    int mid = (l + r) >> 1;    int same = mid -l+1;    for(int i = l; i <= r; ++i) if(tree[dep][i] < sorted[mid]) same--;    int lpos = l;    int rpos = mid + 1;    for(int i = l; i <= r; ++i) {        if(tree[dep][i] < sorted[mid])            tree[dep+1][lpos++] = tree[dep][i];        else if (tree[dep][i] == sorted[mid] && same > 0)            tree[dep+1][lpos++] = tree[dep][i], same--;        else tree[dep+1][rpos++] = tree[dep][i];        toleft[dep][i] = toleft[dep][l-1] + lpos -l;    }    Build(l, mid, dep+1); Build(mid+1,r,dep+1);}int query(int L, int R, int l,int r,int dep, int k) {    if(l == r) return tree[dep][l];    int mid = (L+R)>>1;    int cnt = toleft[dep][r] - toleft[dep][l-1];        //区间有多少数进入左区间    if(cnt >= k) {                                      //递归左树        int newl = L + toleft[dep][l-1]-toleft[dep][L-1];        int newr = newl + cnt - 1;        return query(L,mid, newl, newr, dep+1, k);    }    else {                                              //递归右树        int newr = r + toleft[dep][R] - toleft[dep][r];        int newl = newr - (r-l-cnt);        return query(mid+1, R, newl, newr, dep+1,k-cnt);    }}int main() {    ios::sync_with_stdio(false);    int n, m;    while(cin >> n >> m) {        for(int i = 1; i <= n; ++i) {            cin >> tree[0][i];            sorted[i] = tree[0][i];        }        sort(sorted+1, sorted+n+1);        Build(1,n,0);        while(m--) {            int l, r, k;            cin >> l >> r >> k;            cout << query(1,n,l,r,0,k) << endl;        }    }    return 0;}*/// 主席树, 动态的话就套个树状数组可以了。#include <iostream>#include <algorithm>using namespace std;const int maxn = 100000 + 131;struct Node {    int lpos, rpos;    int w;    Node() {        lpos = rpos = w = 0;    }};Node T[maxn * 30];int a[maxn], b[maxn], p[maxn], root[maxn], sz;bool cmp(int i, int j) {    return a[i] < a[j];}int n, m;void Insert(int &i, int l, int r,int x) {    T[++sz] = T[i]; i = sz;    T[i].w++;    if(l == r) return ;    int m = (l + r) >> 1;    if(x <= m) Insert(T[i].lpos,l,m,x);    else Insert(T[i].rpos, m+1,r,x);}int Query(int i, int j, int l,int r,int k) {    if(l == r) return l;    int t = T[T[j].lpos].w - T[T[i].lpos].w;    int m = (l + r) >> 1;    if(t >= k) return Query(T[i].lpos, T[j].lpos,l,m,k);    else return Query(T[i].rpos,T[j].rpos,m+1,r,k-t);}int main() {    ios::sync_with_stdio(false);    while(cin >> n >> m) {        root[0] = 0;        for(int i = 1; i <= n; ++i) {            cin >> a[i];            p[i] = i;        }        sort(p+1,p+1+n,cmp);        for(int i = 1; i <= n; ++i) b[p[i]] = i;        sz = 0;        for(int i = 1; i <= n; ++i) {            root[i] = root[i-1];            Insert(root[i],1,n,b[i]);        }        while(m--) {            int l, r, k;            cin >> l >> r >> k;            int t = Query(root[l-1],root[r],1,n,k);            cout << a[p[t]] <<endl;        }    }    return 0;}