A 【NOIP2012 day2】疫情控制
|
问题描述
H 国有 n 个城市，这 n 个城市用 n-1 条双向道路相互连通构成一棵树，1 号城市是首都，也是树中的根节点。
H 国的首都爆发了一种危害性极高的传染病。当局为了控制疫情，不让疫情扩散到边境城市（叶子节点所表示的城市），决定动用军队在一些城市建立检查点，使得从首都到边境城市的每一条路径上都至少有一个检查点，边境城市也可以建立检查点。但特别要注意的是，首都是不能建立检查点的。现在，在 H国的一些城市中已经驻扎有军队，且一个城市可以驻扎多个军队。
一支军队可以在有道路连接的城市间移动，并在除首都以外的任意一个城市建立检查点，且只能在一个城市建立检查点。一支军队经过一条道路从一个城市移动到另一个城市所需要的时间等于道路的长度（单位：小时）。请问最少需要多少个小时才能控制疫情。注意：不同的军队可以同时移动。
输入格式
第一行一个整数 n，表示城市个数。
接下来的 n-1 行，每行 3 个整数，u、v、w，每两个整数之间用一个空格隔开，表示从城市 u 到城市 v 有一条长为 w 的道路。数据保证输入的是一棵树，且根节点编号为 1。
接下来一行一个整数 m，表示军队个数。
接下来一行 m 个整数，每两个整数之间用一个空格隔开，分别表示这 m 个军队所驻扎的城市的编号。
输出格式
共一行，包含一个整数，表示控制疫情所需要的最少时间。如果无法控制疫情则输出-1。
样例输入 1
4
1 2 1
1 3 2
3 4 3
2
2 2
样例输出 1
3
样例输入 2
10
2 1 3
2 3 4
1 4 7
5 1 9
6 1 2
4 7 9
7 8 8
9 8 8
1 10 2
5
2 8 5 4 2
样例输出 2
9
样例输入 3
10
1 2 15606
3 1 2777
4 3 2008
1 5 29898
1 6 31457
7 5 4630
6 8 32496
3 9 27660
10 9 29090
4
9 5 10 3
样例输出 3
56750
提示
【输入输出样例说明】
第一支军队在 2 号点设立检查点，第二支军队从 2 号点移动到 3 号点设立检查点，所需时间为 3 个小时。
【数据范围】
保证军队不会驻扎在首都。
对于 20%的数据，2≤ n≤ 10；
对于 40%的数据，2 ≤n≤50，0<w <105；
对于 60%的数据，2 ≤ n≤1000，0<w <106；
对于 80%的数据，2 ≤ n≤10,000；
对于 100%的数据，2≤m≤n≤50,000，0<w <109。
1 #include<cstdio> 2 #include<algorithm> 3 #include<cctype> 4 #define ll long long 5 #define R register int 6 #define For(i, s, n) for (R i = s; i <= n; ++ i) 7 #define Getch() (p1 == p2 && (p2 = (p1 = buf) + fread(buf, 1, 1 << 21, stdin), p1 == p2) ? EOF : *p1 ++) 8 #define maxn 50003 9 #define Log 17 10 using namespace std; 11 int n, u, v, w, m, tot, na, nb; 12 int End[maxn << 1], Len[maxn << 1], Next[maxn << 1], Last[maxn], Army[maxn], F[maxn][Log + 1], vis[maxn], used[maxn], restbj[maxn]; 13 ll ans, l, r, Mid, Dis[maxn][Log + 1], restmin[maxn]; 14 char buf[1 << 23], *p1 = buf, *p2 = buf, obuf[1 << 23], *O = obuf; 15 struct node{ 16 int id; 17 ll rest; 18 }a[maxn], b[maxn]; 19 inline int read() { 20 int x = 0, f = 1; 21 char ch = Getch(); 22 while(!isdigit(ch)){if (ch == '-')f = -1; ch = Getch();} 23 while(isdigit(ch))x = x * 10 + (ch ^ 48), ch = Getch(); 24 return x * f; 25 } 26 void write(long long x) { 27 if (x > 9) write(x / 10); 28 *O ++ = x % 10 + '0'; 29 } 30 void Add(int x, int y, int z) { 31 End[++ tot] = y; 32 Len[tot] = z; 33 Next[tot] = Last[x]; 34 Last[x] = tot; 35 } 36 void DFS(int x, int Las_, ll Gone) { 37 F[x][0] = Las_; 38 Dis[x][0] = Gone; 39 For (i, 1, Log) { 40 F[x][i] = F[F[x][i - 1]][i - 1]; 41 Dis[x][i] = Dis[x][i - 1] + Dis[F[x][i - 1]][i - 1]; 42 } 43 for (R i = Last[x]; i != 0; i = Next[i]) 44 if (End[i] != Las_)DFS(End[i], x, Len[i]); 45 } 46 void ini() { 47 n = read(); 48 scanf("%d", &n); 49 For (i, 1, n - 1) { 50 u = read(); 51 v = read(); 52 w = read(); 53 Add(u, v, w); 54 Add(v, u, w); 55 r += w; 56 } 57 DFS(1, 0, 0); 58 m = read(); 59 scanf("%d", &m); 60 For (i, 1, m)Army[i] = read(); 61 } 62 bool cmp(node x,node y) { 63 return x.rest > y.rest; 64 } 65 int checkok(int x, int las){ //dfs找未被“封死”的子树 66 R bj = 1, i, bbj = 0; 67 if (vis[x])return 1; 68 for (i = Last[x]; i != 0; i = Next[i]) { 69 if (End[i] == las)continue; 70 bbj = 1; 71 if (!checkok(End[i], x)) { 72 bj = 0; 73 if (x == 1)b[++ nb].id = End[i], b[nb].rest = Len[i]; 74 else return 0; 75 } 76 } 77 if(!bbj)return 0; 78 return bj; 79 } 80 bool Check(ll Lim) { 81 R i, j, x, now; 82 ll num; 83 na = nb = 0; 84 For (i, 1, n)vis[i] = restbj[i] = 0; 85 For (i, 1, m)used[i] = 0; 86 For (i, 1, m) { 87 x = Army[i], num = 0; 88 for (j = 17; j >= 0; -- j) //上提军队 89 if (F[x][j] > 1 && num + Dis[x][j] <= Lim) 90 num += Dis[x][j], x = F[x][j]; 91 if (F[x][0] == 1 && num + Dis[x][0] <= Lim){ //处理剩余路程 92 a[++ na].rest = Lim - num - Dis[x][0], a[na].id = i; 93 if (!restbj[x] || a[na].rest < restmin[x]) 94 restmin[x] = a[na].rest, restbj[x] = i; 95 } else vis[x] = 1; 96 } 97 if (checkok(1, 0))return 1; 98 sort(a + 1, a + 1 + na, cmp), sort(b + 1, b + 1 + nb, cmp); //军队在子树间转移 99 now = 1; 100 used[0] = 1; 101 For (i, 1, nb) { 102 if (!used[restbj[b[i].id]]) { 103 used[restbj[b[i].id]] = 1; 104 continue; 105 } 106 while (now <= na && (used[a[now].id] || a[now].rest < b[i].rest))++ now; 107 if (now > na)return 0; 108 used[a[now].id] = 1; 109 } 110 return 1; 111 } 112 void solve() { 113 while (l <= r) { 114 Mid = (l + r) >> 1; 115 if (Check(Mid)) { 116 ans = Mid; 117 r = Mid - 1; 118 } else l = Mid + 1; 119 } 120 write(ans); 121 } 122 int main() { 123 ini(); 124 solve(); 125 fwrite(obuf, O - obuf, 1, stdout); 126 return 0; 127 }
