改编自http://blog.csdn.net/cyberzhg/article/details/7869528
题意:
给出一棵树,每个节点有一个忠诚度和能力值,每次对一个节点进行询问,输出其子孙节点中能力值大于该点的能力值中忠诚度最大的那个,如果没有输出-1。忠诚度的值都不相同。
思路:
区间问题线段树解决,首先要把树形结构转换为线性结构。
基本方法就是将每个点重新编号,儿子节点一定放在父亲节点的左侧,用low[]记录当前节点能访问到的最左节点,一遍深搜就可以确定,这样每次询问就可以确定一个区间。每次查询的区间就是[low[x], x-1],区间记录的是忠诚度最大的数值。
离线预处理答案:按能力值由大到小,新序号从大到小排序,每次先求答案,再加入线段树中,这样就满足了能力值的条件。
#pragma comment(linker, "/STACK:102400000,102400000")
#include<stdio.h>
#include<iostream>
#include<string.h>
#include<algorithm>
#include<map>
using namespace std;
#define M 50012
struct node//ab大的在前,index大的在前
{
int index,su,lo,ab,yl;
/*bool operator < (const node& a)const
{
if(ab==a.ab)
return a.index<index;
return a.ab<ab;
}*/
friend bool operator < (node a, node b)
{
if(a.ab==b.ab)
return a.index>b.index;
return a.ab > b.ab;
}
}staff[M];
struct node1
{
int from,to;
}edge[M];
int head[M],ans[M],tot,now;
void addedge(int sup,int index)
{
edge[tot].to=index;
edge[tot].from=head[sup];
head[sup]=tot++;
}
void dfs(int st)
{
staff[st].su=M;
for(int i=head[st];i!=-1;i=edge[i].from)
{
int v=edge[i].to;
dfs(v);
staff[st].su=min(staff[v].su,staff[st].su);
}
staff[st].index=++now;
staff[st].su=min(staff[st].su,staff[st].index);
}
struct node2
{
int l,r,high;
}tree[M*4];
void bulidtree(int x,int l,int r)
{
tree[x].l=l;
tree[x].r=r;
tree[x].high=-1;
if(l==r)
return;
int mid=(l+r)>>1;
bulidtree(x<<1,l,mid);
bulidtree((x<<1)+1,mid+1,r);
}
void updatetree(int x,int l,int r,int value)
{
if(tree[x].l==l&&tree[x].r==r)
{
tree[x].high=value; return;
}
int mid=(tree[x].l+tree[x].r)>>1;
if(l<=mid)
updatetree(x<<1,l,r,value);
else
updatetree((x<<1)+1,l,r,value);
tree[x].high=max(tree[x<<1].high,tree[(x<<1)+1].high);
}
int querytree(int x,int l,int r)
{
if(l>r)
return -1;
if(tree[x].l==l&&tree[x].r==r)
{
return tree[x].high;
}
int mid=(tree[x].r+tree[x].l)>>1;
if(r<=mid)
return querytree(x<<1,l,r);
else if(l>mid)
return querytree((x<<1)+1,l,r);
else
return max(querytree(x<<1,l,mid),querytree((x<<1)+1,mid+1,r));
}
int main()
{
int t,n,m,i,j,k;
// freopen("data.in", "r", stdin);
// freopen("data1.out", "w", stdout);
scanf("%d",&t);
while(t--)
{
memset(head,-1,sizeof(head));
map<int,int>mymap;
scanf("%d%d",&n,&m);
tot=0;now=0;
for(i=1;i<n;i++)
{
scanf("%d%d%d",&staff[i].su,&staff[i].lo,&staff[i].ab);
staff[i].yl=i;
addedge(staff[i].su,i);
mymap[staff[i].lo]=i;
}
dfs(0);
sort(staff+1,staff+n);
// for(i=1;i<n;i++)
// printf("i=%d ablity=%d index=%d\n",i,staff[i].ab,staff[i].index);
bulidtree(1,1,n);
ans[staff[1].yl]=-1;
updatetree(1,staff[1].index,staff[1].index,staff[1].lo);
for(i=2;i<n;i++)
{
ans[staff[i].yl]=querytree(1,staff[i].su,staff[i].index-1);
updatetree(1,staff[i].index,staff[i].index,staff[i].lo);
}
mymap[-1]=-1;
while(m--)
{
scanf("%d",&k);
printf("%d\n",mymap[ans[k]]);
}
}
return 0;
}