细节比较多的二分+跟LCA倍增差不多的思想
首先有这样一个贪心思路,深度越低的检查点越好,而最长时间和深度具有单调性,即给定时间越长,每个军队能向更浅的地方放置检查点。因此可以考虑二分时间,然后判断军队是否可以放置在控制疫情的地方。
但是有的军队需要先满足自己当前所在的节点,然后此节点如果有多个军队,其他军队跳到1节点,再跳到1节点的其他子树,这里又有一个贪心策略,就是每个军队跳到1的剩余时间:二分的当前时间减去到1节点的距离越大,就要跳到1的另一个子树里据1最远的点,这样才能更好地使得控制疫情。
#include <bits/stdc++.h> #define N 1001001 using namespace std; struct edg { int to, nex, len; }e[N]; struct tem { int id, len; }a[N], b[N]; int n, m, cnt, lin[N], vis[N], vis2[N], army[N], dep[N], fa[N][19], dis[N][19];//fa[i][j]表示军队向上跳j次所到达的位置,dis表示此时所用的时间。 //vis表示该点包括的叶子节点是否被完全覆盖 bool cmp(tem x, tem y) { return x.len < y.len; } inline void add(int f, int t, int l) { e[++cnt].len = l; e[cnt].to = t; e[cnt].nex = lin[f]; lin[f] = cnt; } void dfs(int now, int f) { fa[now][0] = f; dep[now] = dep[f] + 1; for (int i = lin[now]; i; i = e[i].nex) { int to = e[i].to; if (to == f) continue; dis[to][0] = e[i].len; dfs(to, now); } } void bF(int now)//判断该点的所有叶子节点是否全都被覆盖 { int b1 = 1, b2 = 0; if (vis[now]) return; for (int i = lin[now]; i; i = e[i].nex) { int to = e[i].to; if (to == fa[now][0]) continue; b2 = 1; bF(to); if (!vis[to]) b1 = 0; } if (b1 && b2 && now != 1) vis[now] = 1; } bool check(int mid)//使每个军队都尽可能的往上跳,直到距离比mid大才停止。 { memset(vis, 0, sizeof(vis)); memset(vis2, 0, sizeof(vis2)); int tnt1 = 0, tnt2 = 0; for (int i = 1; i <= m; i++) { int u = army[i], dn = 0; for (int j = 18; j >= 0; j--) if (dn + dis[u][j] <= mid && fa[u][j] != 0) dn += dis[u][j], u = fa[u][j]; if (u != 1)//如果此点不为1说明此点可以被覆盖。 vis[u] = 1; else { a[++tnt1].len = mid - dn; int u2 = army[i]; for (int j = 18; j >= 0; j--) if (fa[u2][j] > 1) u2 = fa[u2][j]; a[tnt1].id = u2; } } bF(1); for (int i = lin[1]; i; i = e[i].nex)//枚举深度为2的点 { int to = e[i].to; if (vis[to]) continue;//找到没有完全覆盖完的to; 然后加以覆盖。 选一定要选深度最浅的,所以选择深度为2的,这样选择一定是最优解。 b[++tnt2].id = to, b[tnt2].len = e[i].len; } sort(a + 1, a + 1 + tnt1, cmp);//a的len是剩余的时间 sort(b + 1, b + 1 + tnt2, cmp);//b的len是需要的时间, 他们的时间相加 int j = 1; for (int i = 1; i <= tnt1; i++) { if (!vis[a[i].id]) vis[a[i].id] = 1;//先把此点给覆盖了。 else if (a[i].len >= b[j].len) vis[b[j].id] = 1; while (vis[b[j].id] && j <= tnt2) j++; } if (j > tnt2) return 1; else return 0; } inline void init() { scanf("%d", &n); for (int i = 1, u, v, w; i < n; i++) { scanf("%d%d%d", &u, &v, &w); add(u, v, w); add(v, u, w); } scanf("%d", &m); for (int i = 1; i <= m; i++) scanf("%d", &army[i]); dfs(1, 0); for (int j = 1; j <= 18; j++) for (int i = 1; i <= n; i++) { fa[i][j] = fa[fa[i][j - 1]][j - 1]; dis[i][j] = dis[i][j - 1] + dis[fa[i][j - 1]][j - 1];//dis[i][j]是i向上跳j步所走的路径长度,此叶子节点的意思是最深的节点 } } int main() { init(); int ans, l = 0, r = 70000000; while (l <= r) { int mid = (l + r) >> 1; if (check(mid)) ans = mid, r = mid - 1; else l = mid + 1; } printf("%d", ans); return 0; }