HDU - 4366 Successor DFS区间+线段树

Successor：http://acm.hdu.edu.cn/showproblem.php?pid=4366

参考：https://blog.csdn.net/colin_27/article/details/37811157

题意：

　　有一个公司，每个员工组成一个树形结构，每个员工都有一个能力值和忠诚度。现在问你m次，每次问去掉某个人x，谁可以继任这个职位，要求继任的人是x的子员工，且能力值比x的能力值高，且忠诚度是其他子员工不可想象的（每个员工的忠诚度不同）。

思路：

　　我感觉比较巧妙，但可能是一类题的套路。因为对于每个员工，取代他的人是一定的，所以我们直接预处理出所有的员工答案。首先要dfs，把每个员工 Y 变成一段区间，Y的每个子员工的区间的左端点都在Y的区间中。

对区间总长度建立一个线段树，每个节点表示对应区间的最大忠诚度的值。我们把每个人先按照能力从大到小排序，再一个一个的加到这颗线段树中，这样一来，对于每个点x，线段树中存在的点都是能力大于x的，所以我们就可以询问x对应区间的最大忠诚度，又由于每个人和忠诚度一一对应，映射一下就能得到是哪个人。注意这里对相同能力者的处理。

#include <algorithm>
#include  <iterator>
#include  <iostream>
#include   <cstring>
#include   <cstdlib>
#include   <iomanip>
#include    <bitset>
#include    <cctype>
#include    <cstdio>
#include    <string>
#include    <vector>
#include     <stack>
#include     <cmath>
#include     <queue>
#include      <list>
#include       <map>
#include       <set>
#include   <cassert>
#include <unordered_map>
using namespace std;
//#pragma GCC optimize(3)
//#pragma comment(linker, "/STACK:102400000,102400000")  //c++
#define lson (l , mid , rt << 1)
#define rson (mid + 1 , r , rt << 1 | 1)
#define debug(x) cerr << #x << " = " << x << "\n";
#define pb push_back
#define pq priority_queue

typedef long long ll;
typedef unsigned long long ull;

typedef pair<ll ,ll > pll;
typedef pair<int ,int > pii;
typedef pair<int,pii> p3;

//priority_queue<int> q;//这是一个大根堆q
//priority_queue<int,vector<int>,greater<int> >q;//这是一个小根堆q
#define fi first
#define se second
//#define endl '\n'

#define OKC ios::sync_with_stdio(false);cin.tie(0)
#define FT(A,B,C) for(int A=B;A <= C;++A)  //用来压行
#define REP(i , j , k)  for(int i = j ; i <  k ; ++i)
//priority_queue<int ,vector<int>, greater<int> >que;

const ll mos = 0x7FFFFFFF;  //
const ll nmos = 0x80000000;  //-2147483648
const int inf = 0x3f3f3f3f;
const ll inff = 0x3f3f3f3f3f3f3f3f; //
const int mod = 1e9+;
const double esp = 1e-;
const double PI=acos(-1.0);

template<typename T>
inline T read(T&x){
    x=;int f=;char ch=getchar();
    while (ch<''||ch>'') f|=(ch=='-'),ch=getchar();
    while (ch>=''&&ch<='') x=x*+ch-'',ch=getchar();
    return x=f?-x:x;
}

/*-----------------------showtime----------------------*/
            const int maxn = ;
            int a[maxn],sum[maxn*],ans[maxn];
            int le[maxn],ri[maxn];
            // int hd[maxn*20];
            unordered_map<int,int>hd;
            int n,m,cnt;
            vector<int>mp[maxn];
            struct node {
                int h,ab;
                int id;
            }per[maxn];
            bool cmp(node a,node b){
                return a.ab > b.ab;
            }
            void dfs(int u,int fa){

                le[u] = ++cnt;
                for(int i=; i<mp[u].size(); i++){
                    int v = mp[u][i];
                    if(v!=fa){
                        dfs(v,u);
                    }
                }
                ri[u] = ++cnt;
            }
            void build(int l,int r,int rt){
                sum[rt] = -;
                if(l==r){
                    return;
                }
                int mid = (l+r)>>;
                build(l,mid,rt<<);
                build(mid+,r,rt<<|);
            }
            int query(int l,int r,int rt,int L,int R){
                if(l >= L && r <= R){
                    return sum[rt];
                }
                int mid = (l + r)>>;
                int ans = -;
                if(mid >= L)ans = max(ans, query(l,mid,rt<<,L,R));
                if(mid < R)ans = max(ans, query(mid+,r,rt<<|,L,R));
                return ans;
            }
            void update(int l,int r,int rt,int p,int c){
                if(l==r){
                    // cout<<"p="<<l<<"="<<c<<endl;
                    sum[rt] = c;
                    return;
                }
                int mid = (l + r)>>;
                if(mid >= p)update(l,mid,rt<<,p,c);
                else update(mid+,r,rt<<|,p,c);
                sum[rt] = max(sum[rt<<], sum[rt<<|]);
            }
int main(){
                int cas; scanf("%d", &cas);
                while(cas--){
                    scanf("%d%d", &n, &m);
                    // memset(hd, -1, sizeof(hd));
                    hd.clear();
                    memset(ans,-,sizeof(ans));
                    for(int i=; i<=n; i++)mp[i].clear();
                    for(int i=; i<n; i++){
                        int a,b,c;
                        scanf("%d%d%d", &a, &b, &c);
                        mp[a].pb(i);
                        per[i].h = b; per[i].ab = c; per[i].id = i;
                        hd[b] = i;
                    }
                    cnt = ;
                    dfs(,-);
                    build(,cnt,);
                    per[].id = ;
                    per[].h =per[].ab = -;

                    sort(per,per+n,cmp);
                    for(int i=,j; i<n; i=j){
                        j = i;

                        while(j<n && per[j].ab == per[i].ab){
                            int tmp = query(,cnt,,le[per[j].id],ri[per[j].id]);
                            //cout<<le[per[j].id]<<"   "<<ri[per[j].id]<<endl;
                            if(tmp!=- && hd.count(tmp))ans[per[j].id] = hd[tmp];
                            else ans[per[j].id] = -;
                            j++;
                        }
                        for(int k=i; k<j; k++){
                            update(,cnt,,le[per[k].id],per[k].h);
                        }
                    }
                    while(m--){
                        int x;scanf("%d", &x);
                        printf("%d\n",ans[x]);
                    }
                }
               return ;
}

HDU 4366