HDU 4366 Successor

Successor

Time Limit: 1000ms

Memory Limit: 65536KB

This problem will be judged on HDU. Original ID: 4366
64-bit integer IO format: %I64d Java class name: Main

Sean owns a company and he is the BOSS.The other Staff has one Superior.every staff has a loyalty and ability.Some times Sean will fire one staff.Then one of the fired man’s Subordinates will replace him whose ability is higher than him and has the highest loyalty for company.Sean want to know who will replace the fired man.

Input

In the first line a number T indicate the number of test cases. Then for each case the first line contain 2 numbers n,m (2<=n,m<=50000),indicate the company has n person include Sean ,m is the times of Sean’s query.Staffs are numbered from 1 to n-1,Sean’s number is 0.Follow n-1 lines,the i-th(1<=i<=n-1) line contains 3 integers a,b,c(0<=a<=n-1,0<=b,c<=1000000),indicate the i-th staff’s superior Serial number,i-th staff’s loyalty and ability.Every staff ‘s Serial number is bigger than his superior,Each staff has different loyalty.then follows m lines of queries.Each line only a number indicate the Serial number of whom should be fired.

Output

For every query print a number:the Serial number of whom would replace the losing job man,If there has no one to replace him,print -1.

Sample Input

Sample Output

2

-1

Source

2012 Multi-University Training Contest 7

解题：线段树，dfs序列得出区间，然后将员工按能力降序排序，然后利用线段树查询即可！不过G++爆栈，选择C++交

 #include <iostream>

 #include <vector>

 #include <cstdio>

 #include <algorithm>

 #include <cstring>

 #pragma comment(linker, "/STACK:102400000,102400000")

 using namespace std;

 const int maxn = ;

 struct node {

     int maxv,id;

 } tree[maxn<<];

 void update(int L,int R,int lt,int rt,int val,int id,int v) {

     if(lt <= L && rt >= R) {

         tree[v].maxv = val;

         tree[v].id = id;

         return ;

     }

     int mid = (L + R)>>;

     if(lt <= mid) update(L,mid,lt,rt,val,id,v<<);

     if(rt > mid) update(mid+,R,lt,rt,val,id,v<<|);

     if(tree[v<<].maxv > tree[v<<|].maxv)

         tree[v] = tree[v<<];

     else tree[v] = tree[v<<|];

 }

 node query(int L,int R,int lt,int rt,int v) {

     if(lt <= L && rt >= R) return tree[v];

     int mid = (L + R)>>;

     node a,b;

     a.maxv = -,b.maxv = -;

     a.id = -,b.id = -;

     if(lt <= mid) a = query(L,mid,lt,rt,v<<);

     if(rt > mid) b = query(mid+,R,lt,rt,v<<|);

     if(a.maxv > b.maxv) return a;

     return b;

 }

 vector<int>g[maxn];

 int L[maxn],R[maxn],ret[maxn],tt;

 void dfs(int u) {

     L[u] = tt++;

     for(int i = g[u].size()-; i >= ; --i)

         dfs(g[u][i]);

     R[u] = tt-;

 }

 struct STAFF {

     int id,ability,loyalty;

     bool operator<(const STAFF &t)const {

         if(ability == t.ability) return id < t.id;

         return ability > t.ability;

     }

 } staff[maxn];

 int main() {

     int kase,n,m,fa;

     scanf("%d",&kase);

     while(kase--) {

         scanf("%d%d",&n,&m);

         for(int i = tt = ; i <= n; ++i) g[i].clear();

         for(int i = ; i < n; ++i) {

             scanf("%d%d%d",&fa,&staff[i-].loyalty,&staff[i-].ability);

             g[fa].push_back(i);

             staff[i-].id = i;

         }

         dfs();

         sort(staff,staff+n-);

         memset(tree,-,sizeof tree);

         for(int i = ; i < n-; ++i) {

             node now = query(,R[],L[staff[i].id],R[staff[i].id],);

             ret[staff[i].id] = now.id;

             update(,R[],L[staff[i].id],L[staff[i].id],staff[i].loyalty,staff[i].id,);

         }

         while(m--) {

             scanf("%d",&fa);

             printf("%d\n",ret[fa]);

         }

     }

     return ;

 }