题意:
搜了一下网上几乎都是用优先队列做的我也是醉了。。好吧先说题意:给一列数A[1],A[2],...A[m],和查询序列u[1],u[2],...u[n],查询序列中的每个数u[i],输出A[1],A[2]...A[ u[i] ]中第i个数。
思路:
这题的本质是求动态序列的第k大数,动态序列是指在查询过程中允许对序列进行增加或删除(本题只要求动态增加),线段树可以以 nlogn(n为序列大小)的效率完成动态序列的第k大数的查找工作。
代码:
poj 1442 by sepNINE
#include<iostream>
#include<algorithm>
using namespace std;
const int maxM=30012;
const int maxN=30012;
int A[maxM],tmpA[maxN];
int u[maxN];
struct node{
int l, r;
int c;
}T[3*maxM];
void build(int l,int r,int k){
if(l==r){
T[k].l=l;
T[k].r=r;
T[k].c=0;
return;
}
int m=(l+r)/2;
T[k].l=l;
T[k].r=r;
T[k].c=0;
build(l,m,2*k);
build(m+1,r,2*k+1);
}
void insert(int d,int k)
{
if(T[k].l==T[k].r&&d==T[k].l)
{
T[k].c+=1;
return ;
}
int m=(T[k].l+T[k].r)/2;
if(d<=m)
insert(d,2*k);
else
insert(d,2*k+1);
T[k].c=T[2*k].c+T[2*k+1].c;
}
void search(int d,int k,int& ans)
{
if(T[k].l==T[k].r)
{
ans=T[k].l;
return ;
}
if(d>T[2*k].c)
search(d-T[2*k].c,2*k+1,ans);
else
search(d,2*k,ans);
}
int find(int x,int m)
{
int l=1,r=m+1;
while(l<r){
int mid=(r+l)/2;
if(x>=tmpA[mid])
l=mid+1;
else
r=mid;
}
return l-1;
}
int main()
{
int m,n,i;
scanf("%d%d",&m,&n);
for(i=1;i<=m;++i){
scanf("%d",&A[i]);
tmpA[i]=A[i];
}
for(i=1;i<=n;++i)
scanf("%d",&u[i]);
sort(tmpA+1,tmpA+1+m);
build(1,m,1);
int x=1;
for(i=1;i<=n;++i){
while(x<=u[i]){
insert(find(A[x],m),1);
++x;
}
int ans;
search(i,1,ans);
printf("%d\n",tmpA[ans]);
}
return 0;
}