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 << " "; #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 ' ' #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; //2147483647 const ll nmos = 0x80000000; //-2147483648 const int inf = 0x3f3f3f3f; const ll inff = 0x3f3f3f3f3f3f3f3f; //18 const int mod = 1e9+7; const double esp = 1e-8; const double PI=acos(-1.0); template<typename T> inline T read(T&x){ x=0;int f=0;char ch=getchar(); while (ch<'0'||ch>'9') f|=(ch=='-'),ch=getchar(); while (ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar(); return x=f?-x:x; } /*-----------------------showtime----------------------*/ const int maxn = 50009; int a[maxn],sum[maxn*5],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=0; 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] = -1; if(l==r){ return; } int mid = (l+r)>>1; build(l,mid,rt<<1); build(mid+1,r,rt<<1|1); } int query(int l,int r,int rt,int L,int R){ if(l >= L && r <= R){ return sum[rt]; } int mid = (l + r)>>1; int ans = -1; if(mid >= L)ans = max(ans, query(l,mid,rt<<1,L,R)); if(mid < R)ans = max(ans, query(mid+1,r,rt<<1|1,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)>>1; if(mid >= p)update(l,mid,rt<<1,p,c); else update(mid+1,r,rt<<1|1,p,c); sum[rt] = max(sum[rt<<1], sum[rt<<1|1]); } int main(){ int cas; scanf("%d", &cas); while(cas--){ scanf("%d%d", &n, &m); // memset(hd, -1, sizeof(hd)); hd.clear(); memset(ans,-1,sizeof(ans)); for(int i=0; i<=n; i++)mp[i].clear(); for(int i=1; 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 = 0; dfs(0,-1); build(1,cnt,1); per[0].id = 0; per[0].h =per[0].ab = -1; sort(per,per+n,cmp); for(int i=0,j; i<n; i=j){ j = i; while(j<n && per[j].ab == per[i].ab){ int tmp = query(1,cnt,1,le[per[j].id],ri[per[j].id]); //cout<<le[per[j].id]<<" "<<ri[per[j].id]<<endl; if(tmp!=-1 && hd.count(tmp))ans[per[j].id] = hd[tmp]; else ans[per[j].id] = -1; j++; } for(int k=i; k<j; k++){ update(1,cnt,1,le[per[k].id],per[k].h); } } while(m--){ int x;scanf("%d", &x); printf("%d ",ans[x]); } } return 0; }