题目链接：http://www.lydsy.com/JudgeOnline/problem.php?id=1803
题意：给出一棵 n 个点的树，节点有点权，保证点权两两不同，m 次询问 x 的子树中权值第 k 小的节点编号。
主席树经典题，考虑区间第 k 小在序列上容易解决，现在在树上了。关于 dfs 序有一个性质，一个节点子树中的节点在一段连续的区间里，区间大小为子树大小。
证明是显然的，因为 dfs 的过程一定是搜完这个子树再去考虑外面的情况...
所以在 dfs 序列上建主席树即可啦... 惨遭多次题意杀...
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 |
#include <cmath> #include <queue> #include <cstdio> #include <iomanip> #include <cstdlib> #include <cstring> #include <iostream> #include <algorithm> #define N 100010 #define ll long long using namespace std; inline int read() { int x=0,f=1;char ch=getchar(); while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();} while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();} return x*f; } struct zgzgz { int next,to; }edge[N<<1]; int head[N],cnt=1; void add(int from,int to) { edge[cnt].to=to; edge[cnt].next=head[from]; head[from]=cnt++; } struct zgz { int x,id; }mp[N]; bool cmp(const zgz &a,const zgz &b) {return a.x<b.x;} int n,m,a[N]; int dfn,id[N],tim[N],sz[N]; void dfs(int x,int fa) { id[x]=++dfn,tim[dfn]=x; sz[x]=1; for(int i=head[x];i;i=edge[i].next) { int to=edge[i].to; if(to==fa)continue; dfs(to,x); sz[x]+=sz[to]; } } int tot,sum[N<<5],ch[N<<5][2],root[N]; void update(int x,int &y,int l,int r,int v) { y=++tot; sum[y]=sum[x]+1,ch[y][0]=ch[x][0],ch[y][1]=ch[x][1]; if(l==r)return ; int mid=(l+r)>>1; if(v<=mid)update(ch[x][0],ch[y][0],l,mid,v); else update(ch[x][1],ch[y][1],mid+1,r,v); } int query(int x,int y,int l,int r,int k) { if(l==r)return l; int mid=(l+r)>>1; int t=sum[ch[y][0]]-sum[ch[x][0]]; if(t>=k)return query(ch[x][0],ch[y][0],l,mid,k); else return query(ch[x][1],ch[y][1],mid+1,r,k-t); } int main() { n=read(); for(int i=1;i<=n;i++) mp[i].x=read(),mp[i].id=i; sort(mp+1,mp+n+1,cmp); for(int i=1;i<=n;i++) a[mp[i].id]=i; for(int i=1;i<=n-1;i++) { int x=read(),y=read(); add(x,y),add(y,x); } dfs(1,0); for(int i=1;i<=n;i++) { int x=tim[i]; update(root[i-1],root[i],1,n,a[x]); } m=read(); while(m--) { int x=read(),k=read(); int l=id[x],r=id[x]+sz[x]-1; printf("%d\n",mp[query(root[l-1],root[r],1,n,k)].id); } } |
Comments | NOTHING