Successor HDU - 4366 分块
代码+注释：
1 /* 2 题意： 3 一共有n个人，其中0号是总裁(金字塔顶尖)。后面输入其他n-1个人的信息啊a、b、c，分别代表第i个人的上级是a，他的 4 忠诚度为b，他的能力为c。后面有m次询问。他问你能不能找到一个能力比他高的，且忠诚度最高的人。(注意能力只需要 5 大于此人就可以，不需要最高） 6 7 8 题解： 9 这道题目用的是分块的方法 10 11 首先需要跑一遍dfs来确定每一个人所处于金字塔的位置。从总裁开始向他的下属跑dfs(注意是dfs不是bfs) 12 因为这样我们最后会得到一个pre数组和一个last数组。这两个数组的作用就是来看某个人x他在金字塔中的位置 13 以及他的下属最大的位置(这个位置的值是cnt)。那么pre[x]到last[x]之间的所有人都是x的下属 14 15 然后分块也是按照位置的编号进行分块的，这个样子维护也好维护。分块后会得到一个maxx数组，这个数组的作用就是 16 maxx[x][y]：在第x个块里，能力大于y且忠诚度最高的值，这个p[x][y]存放的是这个值所对应的人 17 18 id数组里面放的是就是，id[x]表示：位置为x的人的在输入中的编号是多少 19 20 具体看代码 21 */ 22 #include<stdio.h> 23 #include<string.h> 24 #include<iostream> 25 #include<algorithm> 26 #include<math.h> 27 #include<vector> 28 using namespace std; 29 const int maxn=50005; 30 const int maxm=sqrt(50000)+5; 31 int n,m,block,num; 32 int belong[maxn],l[maxn],r[maxn],last[maxn],cnt=0,pre[maxn]; 33 int id[maxn],v[maxn],a[maxn],maxx[maxm][maxm],p[maxm][maxm]; 34 struct node 35 { 36 int ability,loyalty,id; 37 node(int l=0,int a=0,int id=0):loyalty(l),ability(a),id(id) {}; 38 bool operator <(const node& r)const 39 { 40 return ability<r.ability; 41 } 42 } c[maxn]; 43 vector<node>b[maxn]; 44 vector<int>g[maxn]; 45 void build() 46 { 47 block = sqrt(n); 48 num = n / block; 49 if(n % block) num++; 50 for(int i = 1; i <= num; i++) 51 { 52 l[i] = (i-1)*block+1; 53 r[i] = i*block; 54 } 55 r[num] = n; 56 for(int i = 1; i <= n; i++) 57 { 58 belong[i] = (i-1)/block+1; 59 } 60 for(int i = 1; i <= num; i++) 61 { 62 for(int j = l[i]; j <= r[i]; j++) 63 { 64 int idd = id[j]; 65 c[j] = node(v[idd], a[idd], idd); 66 b[i].push_back(node(v[idd], a[idd], idd)); 67 } 68 sort(b[i].begin(), b[i].end()); 69 int len = b[i].size(); 70 maxx[i][len] = -1; 71 for(int j = len-1; j >= 0; j--) 72 { 73 if(maxx[i][j+1] < b[i][j].loyalty) 74 { 75 maxx[i][j] = b[i][j].loyalty; 76 p[i][j] = b[i][j].id; 77 } 78 else 79 { 80 maxx[i][j] = maxx[i][j+1]; 81 p[i][j] = p[i][j+1]; 82 } 83 } 84 } 85 } 86 int query(int fa,int x,int y) 87 { 88 int ans=-1,id=-1; 89 if(belong[x]==belong[y]) 90 { 91 for(int i=x; i<=y; ++i) 92 { 93 if(a[fa]<c[i].ability && ans<c[i].loyalty) 94 { 95 ans=c[i].loyalty; 96 id=c[i].id; 97 } 98 } 99 } 100 else 101 { 102 for(int i=x; i<=r[belong[x]]; ++i) 103 { 104 if(a[fa]<c[i].ability && ans<c[i].loyalty) 105 { 106 ans=c[i].loyalty; 107 id=c[i].id; 108 } 109 } 110 111 for(int i=belong[x]+1; i<belong[y]; ++i) 112 { 113 int pos = upper_bound(b[i].begin(), b[i].end(), node(0, a[fa], 0)) - b[i].begin(); 114 int cur = maxx[i][pos]; 115 if(ans < cur) 116 { 117 ans = cur; 118 id = p[i][pos]; 119 } 120 } 121 122 for(int i=l[belong[y]]; i<=y; ++i) 123 { 124 if(a[fa]<c[i].ability && ans<c[i].loyalty) 125 { 126 ans=c[i].loyalty; 127 id=c[i].id; 128 } 129 } 130 } 131 return id; 132 } 133 void init() 134 { 135 for(int i=0; i<=n; ++i) 136 { 137 b[i].clear(); 138 g[i].clear(); 139 } 140 cnt=0; 141 } 142 void dfs(int root,int fa) 143 { 144 pre[root]=++cnt; 145 id[cnt]=root; 146 int len=g[root].size(); 147 for(int i=0; i<len; ++i) 148 { 149 int v=g[root][i]; 150 if(v!=fa) dfs(v,root); 151 } 152 last[root]=cnt; 153 } 154 int main() 155 { 156 int t; 157 scanf("%d",&t); 158 while(t--) 159 { 160 scanf("%d%d",&n,&m); 161 init(); 162 for(int i=1; i<=n-1; ++i) 163 { 164 int x; 165 scanf("%d%d%d",&x,&v[i],&a[i]); 166 g[x].push_back(i); 167 } 168 dfs(0,0); 169 build(); 170 while(m--) 171 { 172 int x; 173 scanf("%d",&x); 174 printf("%d\n",query(x,pre[x],last[x])); 175 } 176 } 177 return 0; 178 }
