K-th Number POJ - 2104

K-th Number POJ - 2104

You are working for Macrohard company in data structures department. After failing your previous task about key insertion you were asked to write a new data structure that would be able to return quickly k-th order statistics in the array segment. 
That is, given an array a[1...n] of different integer numbers, your program must answer a series of questions Q(i, j, k) in the form: "What would be the k-th number in a[i...j] segment, if this segment was sorted?" 
For example, consider the array a = (1, 5, 2, 6, 3, 7, 4). Let the question be Q(2, 5, 3). The segment a[2...5] is (5, 2, 6, 3). If we sort this segment, we get (2, 3, 5, 6), the third number is 5, and therefore the answer to the question is 5. 

Input

The first line of the input file contains n --- the size of the array, and m --- the number of questions to answer (1 <= n <= 100 000, 1 <= m <= 5 000). 
The second line contains n different integer numbers not exceeding 10⁹ by their absolute values --- the array for which the answers should be given. 
The following m lines contain question descriptions, each description consists of three numbers: i, j, and k (1 <= i <= j <= n, 1 <= k <= j - i + 1) and represents the question Q(i, j, k). 

Output

For each question output the answer to it --- the k-th number in sorted a[i...j] segment. 

Sample Input

7 3
1 5 2 6 3 7 4
2 5 3
4 4 1
1 7 3

Sample Output

5
6
3

Hint

This problem has huge input,so please use c-style input(scanf,printf),or you may got time limit exceed.

 

 

题意：求区间第k大

题解：主席树的板子

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<sstream>
#include<cmath>
#include<stack>
#include<cstdlib>
#include <vector>
#include<queue>
using namespace std;
 
#define ll long long
#define llu unsigned long long
#define INF 0x3f3f3f3f
#define PI acos(-1.0)
const int maxn =  1e5+;
const int  mod = 1e9+;
 
int n,q,m,tot;
int a[maxn],t[maxn],T[maxn],lson[maxn*],rson[maxn*],c[maxn*];
 
void Init_hash()
{
    for(int i=;i<=n;i++)
        t[i] = a[i];
     sort(t+,t++n);
     m = unique(t+,t++n)-t-;
}
 
int build(int l,int r)
{
    int root = tot++;
    c[root] = ;
    if (l != r)
    {
        int mid = (l+r) >> ;
        lson[root] = build(l,mid);
        rson[root] = build(mid+,r);
    }
    return root;
}
int Hash(int x)
{
    return lower_bound(t+,t++m,x)-t;
}
int update(int root,int pos,int val)
{
    int newroot = tot ++,tmp = newroot;
    c[newroot] = c[root] + val;
    int l = ,r = m;
    while(l <r)
    {
        int mid = (l+r)>>;
        if(pos <= mid)
        {
            lson[newroot] = tot++;
            rson[newroot] = rson[root];
            newroot = lson[newroot];
            root = lson[root];
            r = mid;
        }
        else
        {
            rson[newroot] = tot ++;
            lson[newroot] = lson[root];
            newroot = rson[newroot];
            root = rson[root];
            l = mid + ;
        }
        c[newroot] = c[root] + val;
    }
    return tmp;
 
}
int query(int left_root,int right_root,int k)
{
    int l = ,r = m;
    while(l < r)
    {
        int mid = (l+r) >> ;
        if(c[lson[left_root]] - c[lson[right_root]] >= k)
        {
            r = mid;
            left_root = lson[left_root];
            right_root = lson[right_root];
        }
        else
        {
            l = mid + ;
            k -= c[lson[left_root]] - c[lson[right_root]];
            left_root = rson[left_root];
            right_root = rson[right_root];
        }
    }
    return l;
}
int main()
{
    scanf("%d%d",&n,&q);
    tot = ;
    for(int i=;i<=n;i++)
        scanf("%d",&a[i]);
    Init_hash();
    T[n+] = build(,m);
    for(int i=n;i;i--)
    {
        int pos = Hash(a[i]);
        T[i] = update(T[i+],pos,);
    }
    while(q--)
    {
        int l,r,k;
        scanf("%d%d%d",&l,&r,&k);
        printf("%d\n",t[query(T[l],T[r+],k)]);
    }
}

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<stack>
#include<cstdlib>
#include<queue>
#include<set>
#include<string.h>
#include<vector>
#include<deque>
#include<map>
using namespace std;
#define INF 0x3f3f3f3f3f3f3f3f
#define inf 0x3f3f3f3f
#define eps 1e-4
#define bug printf("*********\n")
#define debug(x) cout<<#x"=["<<x<<"]" <<endl
typedef long long LL;
typedef long long ll;
const int maxn = 2e5 + ;
const int mod = ;
int n,m,a[maxn],root[maxn],cnt;
vector<int>v;
struct node{
    int l,r,sum;
}T[maxn * ];
int getid(int x) {
    return lower_bound(v.begin(),v.end(),x) - v.begin() + ;
}
void update(int l,int r,int &x,int y,int pos) {
    T[++cnt] = T[y],T[cnt].sum ++,x = cnt;
    if(l == r) return;
    int mid = (l + r) / ;
    if(mid >= pos) update(l,mid,T[x].l,T[y].l,pos);
    else update(mid + ,r,T[x].r,T[y].r,pos);
}
int query(int l,int r,int x,int y,int k) {
    if(l == r) return l;
    int mid = (l + r) / ;
    int sum = T[T[y].l].sum - T[T[x].l].sum;
    if(sum >= k) return query(l,mid,T[x].l,T[y].l,k);
    else return query(mid + ,r,T[x].r,T[y].r,k - sum);
}
void init() {
    cnt = ;
    memset(root,,sizeof root);
}
int main()
{
    init();
    scanf("%d %d",&n,&m);
    for (int i = ; i <= n; ++i) {
        scanf("%d",&a[i]);
        v.push_back(a[i]);
    }
    sort(v.begin(),v.end());
    v.erase(unique(v.begin(),v.end()),v.end());
    for (int i = ; i <= n; ++i)
        update(,n,root[i],root[i - ],getid(a[i]));
    for (int i = ; i <= m; ++i) {
        int x,y,k;
        scanf("%d %d %d",&x, &y, &k);
        printf("%d\n",v[query(,n,root[x - ],root[y],k) - ]);
    }
}

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<stack>
#include<cstdlib>
#include<queue>
#include<set>
#include<string.h>
#include<vector>
#include<deque>
#include<map>
using namespace std;
#define INF 0x3f3f3f3f3f3f3f3f
#define inf 0x3f3f3f3f
#define eps 1e-4
#define bug printf("*********\n")
#define debug(x) cout<<#x"=["<<x<<"]" <<endl
typedef long long LL;
typedef long long ll;
const int maxn = 2e5 + ;
const int mod = ;
int n,m,a[maxn],root[maxn],cnt;
/*
 root：代表每个历史版本线段树的根节点位置
 cnt：用作开辟新的树节点
 */
vector<int>v;
struct node{
    int l,r,sum;
}T[maxn * ];  //线段树区间统计，sum代表在这个区间数的个数
int getid(int x) {      //获取离散数组后的下标
    return lower_bound(v.begin(),v.end(),x) - v.begin() + ;
}
/*
 update函数：
 y代表前一棵树的节点位置，x是后面的节点位置
 */
void update(int l,int r,int &x,int y,int pos) {
    //&x 引用节点指针扩展新节点
    T[++cnt] = T[y],T[cnt].sum ++,x = cnt;  //新开节点，将需要修改的树节点复制到新开辟节点，改变自己的sum
    if(l == r) return;
    int mid = (l + r) / ;
    if(mid >= pos) update(l,mid,T[x].l,T[y].l,pos); //节点左边
    else update(mid + ,r,T[x].r,T[y].r,pos);      //节点右边
}
int query(int l,int r,int x,int y,int k) {
    if(l == r) return l;
    int mid = (l + r) / ;
    int sum = T[T[y].l].sum - T[T[x].l].sum;    //两颗线段树sum做差
    if(sum >= k) return query(l,mid,T[x].l,T[y].l,k);   //如果当前的结点个数sum比k要大的话，说明第k大的结点在左子树当中，就去遍历左子树
    else return query(mid + ,r,T[x].r,T[y].r,k - sum); //如果当前的结点个数sum比k要小的话，说明第k大的结点在右子树当中，而左子树的结点个数是sum个，那就去找右子树中的第k-sum的数
}
void init() {
    cnt = ;
    memset(root,,sizeof root);
}
int main()
{
    init();
    scanf("%d %d",&n,&m);
    for (int i = ; i <= n; ++i) {
        scanf("%d",&a[i]);
        v.push_back(a[i]);      //离散化数组
    }
    sort(v.begin(),v.end());    //离散化数组
    v.erase(unique(v.begin(),v.end()),v.end()); //离散化数组
 
    for (int i = ; i <= n; ++i) {
        update(, n, root[i], root[i - ], getid(a[i]));
    }
    for (int i = ; i <= m; ++i) {
        int x,y,k;
        scanf("%d %d %d",&x, &y, &k);
        printf("%d\n",v[query(,n,root[x - ],root[y],k) - ]);
    }
}