POJ 2104 K-th number

Time Limit: 20000MS		Memory Limit: 65536K
Total Submissions: 56956		Accepted: 19644
Case Time Limit: 2000MS

Description

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.

Source

Northeastern Europe 2004, Northern Subregion

【中文题意】

给出一个长度为n的序列a1~an，有m次询问(x,y,k)，每次询问a[x]~a[y]内的第k小数。

输入第一行为n,m，第二行为a1~an，接下来m行是m个(x,y,k)。

由于数据较大，请使用C风格的输入输出。

1<=n<=100000,1<=m<=5000

思路

主席树+权值线段树

代码实现

#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=1e5+10;
int n,k,sz,ts;
int a,b,c;
int s[maxn],hs[maxn],tt[maxn];
struct tree{int s,l,r,mid,lp,rp;}t[maxn<<5];
void build(int l,int r,int k){
    t[k].l=l,t[k].r=r;
    if(l==r) return;
    t[k].mid=l+r>>1,t[k].lp=++ts,t[k].rp=++ts;
    build(l,t[k].mid,t[k].lp);
    build(t[k].mid+1,r,t[k].rp);
}
void put(int l,int r,int k,int nk,int p){
    t[nk]=(tree){t[k].s+1,t[k].l,t[k].r,t[k].mid};
    if(l==r) return;
    if(p<=t[nk].mid){
        t[nk].lp=++ts,t[nk].rp=t[k].rp;
        put(l,t[nk].mid,t[k].lp,t[nk].lp,p);
    }
    else{
        t[nk].rp=++ts,t[nk].lp=t[k].lp;
        put(t[nk].mid+1,r,t[k].rp,t[nk].rp,p);
    }
}
int search(int l,int r,int k,int nk,int v){
    if(l==r) return l;
    int w=t[t[nk].lp].s-t[t[k].lp].s;
    if(v<=w) return search(l,t[k].mid,t[k].lp,t[nk].lp,v);
    else return search(t[k].mid+1,r,t[k].rp,t[nk].rp,v-w);
}
int main(){
    freopen("kthnumber.in","r",stdin);
    freopen("kthnumber.out","w",stdout);
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++){
        scanf("%d",&s[i]);
        hs[i]=s[i];
    }
    sort(hs+1,hs+n+1);
    sz=unique(hs+1,hs+n+1)-hs-1;
    build(1,sz,tt[0]);
    for(int i=1;i<=n;i++){
        int pos=lower_bound(hs+1,hs+sz+1,s[i])-hs;
        tt[i]=++ts;
        put(1,sz,tt[i-1],tt[i],pos);
        
    }
    for(int i=1;i<=k;i++){
        scanf("%d%d%d",&a,&b,&c);
        printf("%d
",hs[search(1,n,tt[a-1],tt[b],c)]);
    }
    return 0;
}

仍然不太会，这并不是板子（这份代码应该有坑）。//虽然能AT.

2104

Accepted

47820K

1922MS

G++

1412B

2017-06-07 19:59:16

划分树

我听说划分树是把快速排序的流程存储在线段树上；

代码实现

#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=1e5+10;
int n,q,a,b,c;
int s[maxn],str[20][maxn],cnt[20][maxn];
//s记录排序后的，str记录每一层的排序结果 ，cnt[i][j]表示第i层前j个数中有多少个数进入左子树；
int lt[maxn<<2],rt[maxn<<2],md[maxn<<2];
void build(int k,int l,int r,int dp){
    lt[k]=l,rt[k]=r,md[k]=l+r>>1;
    if(l==r) return;
    int mid=md[k],v=s[mid],lmid=mid-l+1,L=l,R=mid+1;
    for(int i=l;i<=r;i++) if(str[dp][i]<v) lmid--;//lmid表示左子树中还需要多少个中值；
    for(int i=l;i<=r;i++){
        if(i==l) cnt[dp][i]=0;
        else cnt[dp][i]=cnt[dp][i-1];
        if(str[dp][i]<v||(str[dp][i]==v&&lmid>0)){//构建下一层的左子树； 
            str[dp+1][L++]=str[dp][i],cnt[dp][i]++;
            if(str[dp][i]==v) lmid--;
        }
        else str[dp+1][R++]=str[dp][i];
    }
    build(k<<1,l,mid,dp+1);
    build(k<<1|1,mid+1,r,dp+1);
}
int query(int k,int l,int r,int v,int dp){
    if(l==r) return str[dp][l];
    int s1,s2;
    if(lt[k]==l) s1=0;//s1为(lt[k],l-1)中分到左子树的个数；
    else s1=cnt[dp][l-1];
    s2=cnt[dp][r]-s1;//s2为(l,r)中分到左子树的个数；
    if(v<=s2) return query(k<<1,lt[k]+s1,lt[k]+s1+s2-1,v,dp+1);
    //左子树的数量大于k，递归左子树；
    int b1=l-lt[k]-s1;//b1为(lt[k],l-1)中分到右子树的个数； 
    int b2=r-l-s2+1;//b2为(l,r)中分到右子树的个数； 
    return query(k<<1|1,md[k]+b1+1,md[k]+b1+b2,v-s2,dp+1);
}
int main(){
    scanf("%d%d",&n,&q);
    for(int i=1;i<=n;i++){
        scanf("%d",&str[1][i]);
        s[i]=str[1][i];
    }
    sort(s+1,s+n+1);
    build(1,1,n,1);
    while(q--){
        scanf("%d%d%d",&a,&b,&c);
        printf("%d
",query(1,a,b,c,1));
    }
    return 0;
}

2104

Accepted

17572K

1313MS

G++

1396B

2017-06-07 21:25:11

归并树

 我听说归并树就是把归并排序的流程存储在树上；

#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;
const int maxn=3e5+10;
vector<int>t[maxn<<2];
int n,q,a,b,c;
void build(int k,int l,int r){
    if(l==r){
        scanf("%d",&a);
        t[k].clear();
        t[k].push_back(a);
        return;
    }
    int mid=l+r>>1,ls=k<<1,rs=ls|1;
    build(ls,l,mid);
    build(rs,mid+1,r);
    t[k].resize(r-l+1);//调整容器t[]的大小为r-l+1； 
    merge(t[ls].begin(),t[ls].end(),t[rs].begin(),t[rs].end(),t[k].begin());//将两个儿子的数列合并； 
}
int query(int k,int l,int r,int al,int ar,int v){
    if(al==l&&ar==r) return upper_bound(t[k].begin(),t[k].end(),v)-t[k].begin();
    int mid=l+r>>1,ls=k<<1,rs=ls|1;
    if(ar<=mid) return query(ls,l,mid,al,ar,v);
    if(al>mid) return query(rs,mid+1,r,al,ar,v);
    return query(ls,l,mid,al,mid,v)+query(rs,mid+1,r,mid+1,ar,v);
}
int main(){
    scanf("%d%d",&n,&q);
    build(1,1,n);
    while(q--){
        scanf("%d%d%d",&a,&b,&c);
        int l=-1,r=n-1,mid;
        while(r-l>1){
            mid=l+r>>1;
            if(query(1,1,n,a,b,t[1][mid])>=c) r=mid;
            else l=mid;
        }
        printf("%d
",t[1][r]);
    }
    return 0;
}

2104

Accepted

26840K

6266MS

G++

1012B

2017-06-08 09:35:11

分块+二分

#include<cmath>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxm=3e3+10;
const int maxn=3e5+10;
int n,q,a,b,c;
int bks,bs;
int s[maxn],ss[maxn];
int bk[maxm][maxm],len[maxm],bkl[maxm],bkr[maxm];
void build(int n){
    bs=n/((int)(sqrt(n*log2(n)))+1);
    if(n%bs) bks=n/bs+1;
    else bks=n/bs;
    for(int i=1;i<=n;i++){
        if(i%bks==1) bkl[i/bks+1]=i;
        if(i%bks) bk[i/bks+1][++len[i/bks+1]]=s[i];
        if(i%bks==0) bkr[i/bks]=i,bk[i/bks][++len[i/bks]]=s[i];
    }
    bkr[bs]=n;
    for(int i=1;i<=bs;i++) sort(bk[i]+1,bk[i]+len[i]+1);
}
int find(int i,int v){
    int l,r,ans;
    l=ans=0,r=len[i]+1;
    while(r-l>0){
        int mid=l+r>>1;
        if(bk[i][mid]<=v) ans=mid,l=mid+1;
        else r=mid;
    }
    return ans;
}
bool check(int l,int r,int k,int v){
    int ans=0;
    for(int i=1;i<=bs;i++){
        if(bkl[i]>=l&&bkr[i]<=r) ans+=find(i,v);
        else if(bkl[i]<=l&&bkr[i]>=r) for(int j=l;j<=r;j++){if(s[j]<=v) ans++;}
        else if(bkl[i]<=l&&bkr[i]>=l) for(int j=l;j<=bkr[i];j++){if(s[j]<=v) ans++;}
        else if(bkl[i]<=r&&bkr[i]>=r) for(int j=bkl[i];j<=r;j++){if(s[j]<=v) ans++;}
    }
    return ans<k;
}
int div(int L,int R,int k){
    int l,r;
    l=0,r=n;
    while(r-l>1){
        int mid=l+r>>1;
        if(check(L,R,k,ss[mid])) l=mid;
        else r=mid;
    }
    return ss[r];
}
int main(){
    scanf("%d%d",&n,&q);
    for(int i=1;i<=n;i++){
        scanf("%d",&s[i]);
        ss[i]=s[i];
    }
    sort(ss+1,ss+n+1);
    build(n);
    while(q--){
        scanf("%d%d%d",&a,&b,&c);
        printf("%d
",div(a,b,c));
    }
    return 0;
}

2104

Accepted

1868K

8985MS

G++

1571B

2017-06-08 08:26:44

个人偏好主席树。//只会写主席树。。。

显然划分树最快而且其空间也是最小的（如果归并树的STL减益不是太大）；

主席树次之，但是空间使用量最大；

归并树用了大量的空间换取时间；

分块看起来最暴力，时间上也确实如此；