社区讨论
求调
SP1487PT07J - Query on a tree III参与者 2已保存回复 2
讨论操作
快速查看讨论及其快照的属性,并进行相关操作。
- 当前回复
- 2 条
- 当前快照
- 1 份
- 快照标识符
- @m1wxqf0w
- 此快照首次捕获于
- 2024/10/06 09:59 去年
- 此快照最后确认于
- 2025/11/04 17:54 4 个月前
RT
可持久化线段树维护第k小
用dfs序将树变为区间
C#include <bits/stdc++.h>
using namespace std;
#define int long long
const int N=100005;
map<int,int> mp;
struct Node
{
int v;
int l,r;
}t[N<<8];
int a[N];
int dfn[N];
int rt[N<<2];
int cnt;
int lsh[N<<2];
int tot;
int n,m;
int node[N<<2];
vector<int> G[N];
int siz[N];
void dfs(int x,int fa)
{
siz[x]++;
dfn[x]=++cnt;
for(int v:G[x])
{
if(v!=fa)
{
siz[x]+=siz[v];
dfs(v,x);
}
}
return;
}
int query(int o1,int o2,int l,int r,int q)
{
if(l==r) return l;
int mid=l+r>>1;
int tmp=t[t[o2].l].v-t[t[o1].l].v;
if(tmp>=q) return query(t[o1].l,t[o2].l,l,mid,q);
return query(t[o1].r,t[o2].r,mid+1,r,q-tmp);
}
void pushup(int o)
{
t[o].v=t[t[o].l].v+t[t[o].r].v;
}
void change(int lsto,int &o,int l,int r,int q,int v)
{
if(!o) o=++cnt;
if(l==r)
{
t[o].v+=v;
return;
}
int mid=l+r>>1;
if(q<=mid)
{
t[o].r=t[lsto].r;
t[o].l=++cnt;
t[t[o].l]=t[t[lsto].l];
change(t[lsto].l,t[o].l,l,mid,q,v);
}
else
{
t[o].l=t[lsto].l;
t[o].r=++cnt;
t[t[o].r]=t[t[lsto].r];
change(t[lsto].r,t[o].r,mid+1,r,q,v);
}
pushup(o);
}
signed main()
{
cin>>n;
for(int i=1;i<=n;i++)
{
scanf("%lld",&a[i]);
mp[a[i]]=i;
}
for(int i=1;i<n;i++)
{
int u,v;
scanf("%lld%lld",&u,&v);
G[u].push_back(v);
G[v].push_back(u);
}
dfs(1,-1);
cnt=0;
for(int i=1;i<=n;i++)
{
//cout<<i<<" : "<<dfn[i]<<endl;
node[dfn[i]]=a[i];
lsh[dfn[i]]=node[dfn[i]];
}
sort(lsh+1,lsh+n+1);
tot=unique(lsh+1,lsh+n+1)-lsh-1;
for(int i=1;i<=n;i++)
{
node[i]=lower_bound(lsh+1,lsh+tot+1,node[i])-lsh;
change(rt[i-1],rt[i],1,tot,node[i],1);
}
cin>>m;
for(int i=1;i<=m;i++)
{
int l,q;
scanf("%lld%lld",&l,&q);
printf("%lld\n",mp[lsh[query(rt[dfn[l]-1],rt[dfn[l]+siz[l]-1],1,tot,q)]]);
}
return 0;
}
回复
共 2 条回复,欢迎继续交流。
正在加载回复...