考虑满足什么条件的水果(x1,y1)会落在指定盘子(x2,y2)上:
如果x2,y2为祖先孩子关系,设x2是y2的祖先,t是x2到y2路径上第二个点那么x1,y1一定满足x1在y2子树中,y1在x2子树外。
在子树中可以表示为dfs序中一段,在子树外可以表示为两段。
否则x1一定在x2子树中,y1一定在y2子树中。
这样问题就转化成平面上一些矩形,每个矩形有权值,一些询问,求包含一个点的权值第k小的矩形。
离线下来,把一维的矩形坐标拆成在x1处+1,x2+1处-1。排一下序。然后就变成了一个二维问题。
需要普通线段树维护位置,权值线段树维护第K大。
由于外层树结构不支持区间修改,因此需要权值线段树套普通线段树。
#include <bits/stdc++.h>
using namespace std;
#define N 41000
#define M 21000000
int n,m,q,tot,cnt,top;
int lq,rq,type;
int head[N],nex[N<<1],to[N<<1],deep[N],fa[N][21];
int pos[N],beg[N],en[N],ans[N],st[N];
struct node
{
int pos,beg,en,val,type;
friend bool operator < (const node &r1,const node &r2)
{return r1.pos<r2.pos;};
}a[N<<3];
struct poi
{
int u,v,K,pos;
void read(){scanf("%d%d%d",&u,&v,&K);}
friend bool operator < (const poi &r1,const poi &r2)
{return r1.u<r2.u;};
}p[N];
void add(int x,int y)
{
tot++;
nex[tot]=head[x];head[x]=tot;
to[tot]=y;
}
void dfs(int x,int y)
{
fa[x][0]=y;
for(int i=1;i<=20;i++)
fa[x][i]=fa[fa[x][i-1]][i-1];
deep[x]=deep[y]+1;
pos[x]=++cnt;beg[x]=cnt;
for(int i=head[x];i;i=nex[i])
if(to[i]!=y)dfs(to[i],x);
en[x]=cnt;
}
int up(int x,int y)
{
for(int i=0;i<=20;i++)
if(y>>i&1)x=fa[x][i];
return x;
}
void Ins(int l1,int r1,int l2,int r2,int v)
{
a[++cnt]=(node){l1,l2,r2,v,1};
a[++cnt]=(node){r1+1,l2,r2,v,-1};
}
void ins(int l1,int r1,int l2,int r2,int v)
{
if(l2>r2)return;
Ins(l1,r1,l2,r2,v);Ins(l2,r2,l1,r1,v);
}
struct seg_tree
{
int ch[M][2],num[M],cnt;
void insert(int l,int r,int &now)
{
if(!now)now=++cnt;
if(lq<=l&&r<=rq)
{num[now]+=type;return;}
int mid=(l+r)>>1;
if(mid>=lq)insert(l,mid,ch[now][0]);
if(mid<rq) insert(mid+1,r,ch[now][1]);
}
int query(int l,int r,int now,int pos)
{
if(l==r)return num[now];
int mid=(l+r)>>1;
if(mid>=pos)return num[now]+query(l,mid,ch[now][0],pos);
else return num[now]+query(mid+1,r,ch[now][1],pos);
}
}tr2;
struct val_tree
{
int root[N<<2];
void insert(int l,int r,int now,int v)
{
tr2.insert(1,n,root[now]);
if(l==r)return;
int mid=(l+r)>>1;
if(mid>=v)insert(l,mid,now<<1,v);
else insert(mid+1,r,now<<1|1,v);
}
int query(int l,int r,int now,int rem,int pos)
{
if(l==r)return st[l];
int mid=(l+r)>>1;
int sum=tr2.query(1,n,root[now<<1],pos);
if(sum>=rem)return query(l,mid,now<<1,rem,pos);
else return query(mid+1,r,now<<1|1,rem-sum,pos);
}
}tr1;
int main()
{
scanf("%d%d%d",&n,&m,&q);
for(int i=1,x,y;i<n;i++)
{
scanf("%d%d",&x,&y);
add(x,y);add(y,x);
}
dfs(1,0);cnt=0;
for(int i=1,x,y,v;i<=m;i++)
{
scanf("%d%d%d",&x,&y,&v);st[++top]=v;
if(deep[x]>deep[y])swap(x,y);
if(beg[x]<=beg[y]&&en[y]<=en[x])
{
int t=up(y,deep[y]-deep[x]-1);
ins(beg[y],en[y],1,beg[t]-1,v);
ins(beg[y],en[y],en[t]+1,n,v);
}
else ins(beg[x],en[x],beg[y],en[y],v);
}
sort(st+1,st+1+top);
top=unique(st+1,st+1+top)-st-1;
for(int i=1;i<=cnt;i++)
a[i].val=lower_bound(st+1,st+1+top,a[i].val)-st;
sort(a+1,a+1+cnt);
for(int i=1;i<=q;i++)
{
p[i].read(),p[i].pos=i;
p[i].u=pos[p[i].u];p[i].v=pos[p[i].v];
}
sort(p+1,p+1+q);
for(int i=1,now=1;i<=q;i++)
{
while(now<=cnt&&a[now].pos<=p[i].u)
{
lq=a[now].beg,rq=a[now].en;type=a[now].type;
tr1.insert(1,top,1,a[now].val);now++;
}
ans[p[i].pos]=tr1.query(1,top,1,p[i].K,p[i].v);
}
for(int i=1;i<=q;i++)
printf("%d\n",ans[i]);
return 0;
}