思路
显而易见,答案具有单调性,时间越长,越容易控制疫情.所以很容易想到二分时间.
首先,占据一个点肯定不如占据这个点的父亲,因此时间内到达不了
号节点的尽力往上跳就可以了(暴力跳可不行,需要使用倍增).
还有一些军队可能会经过号节点到达其他
号节点的儿子节点.
先来证明一个结论:若些军队的集合不经过
号节点能到达
号节点的儿子节点,并且其不能到达
号节点又回到
,还有其他不经过
号节点的军队还未占据
,
中的某一支军队(当然是还能走的路程最小的那支)来占据
号节点不会更劣.
这个其实不难证明,若来自的另一个儿子节点
的军队
来占据
,而
的军队都穿过
去占据其他节点,说明
在节点
时还能走的路程大于等于
,而
中的军队还能走的路程都小于
,得到的解不可能更优.
接下来还要处理那些经过的军队与还需要占据的
的子节点如何配对.
直接使用贪心,军队还能走的路程从大到小排序,子节点按照与距离从大到小排序,按此配对即可.如果不能完成配对,说明
时间不够.这样复杂度为
,一个
是二分的,还有一个是往上跳时倍增的.
代码
#include<bits/stdc++.h>
using namespace std;
#define i64 long long
#define fp( i, b, e ) for ( int i(b), I(e); i <= I; ++i )
#define fd( i, b, e ) for ( int i(b), I(e); i >= I; --i )
#define go( i, b ) for ( int i(b), v(to[i]); i; v = to[i = nxt[i]] )
template<typename T> inline void cmax( T &x, T y ){ x < y ? x = y : x; }
template<typename T> inline void cmin( T &x, T y ){ y < x ? x = y : x; }
#define getchar() ( p1 == p2 && ( p1 = bf, p2 = bf + fread( bf, 1, 1 << 21, stdin ), p1 == p2 ) ? EOF : *p1++ )
char bf[1 << 21], *p1(bf), *p2(bf);
template<typename T>
inline void read( T &x ){ char t(getchar()), flg(0); x = 0;
for ( ; !isdigit(t); t = getchar() ) flg = t == '-';
for ( ; isdigit(t); t = getchar() ) x = x * 10 + ( t & 15 );
flg ? x = -x : x;
}
clock_t t_bg, t_ed;
const int MAXN = 50005;
int N, M;
int hd[MAXN], nxt[MAXN<<1], to[MAXN<<1], val[MAXN<<1], tot;
inline void addedge( int x, int y, int z ){
nxt[++tot] = hd[x], hd[x] = tot, to[tot] = y, val[tot] = z,
nxt[++tot] = hd[y], hd[y] = tot, to[tot] = x, val[tot] = z;
}
int fa[MAXN][20], dep[MAXN], w[MAXN], n, m; i64 s[MAXN], h[MAXN], g[MAXN];
bool b[MAXN];
vector<i64> v[MAXN];
void DFS( int u ){
dep[u] = dep[fa[u][0]] + 1;
fp( i, 1, 15 ) fa[u][i] = fa[fa[u][i - 1]][i - 1];
go( i, hd[u] ) if ( v != fa[u][0] ) fa[v][0] = u, s[v] = s[u] + val[i], DFS(v);
}
void dfs( int u ){
if ( !b[u] ){
if ( nxt[hd[u]] ) b[u] = 1;
go( i, hd[u] ) if ( v != fa[u][0] ) dfs(v), b[u] &= b[v];
}
if ( dep[u] == 2 && v[u].size() ){
if ( b[u] ) fp( i, 0, v[u].size() - 1 ) h[++n] = v[u][i] - s[u];
else{ b[u] = 1, sort( v[u].begin(), v[u].end() ); fp( i, 1, (int)v[u].size() - 1 ) h[++n] = v[u][i] - s[u]; }
}
} bool cmp( int x, int y ){ return x > y; }
bool check( i64 S ){
n = m = 0;
memset( b, 0, sizeof b );
fp( i, 1, N ) v[i].clear();
fp( i, 1, M ){
int x(w[i]), t(x); i64 y(S);
fd( j, 15, 0 ) if ( fa[t][j] > 1 && s[x] - s[fa[t][j]] <= y ) t = fa[t][j];
y -= s[x] - s[t];
if ( dep[t] == 2 && y >= (s[t] << 1) ) h[++n] = y - s[t];
else if ( dep[t] == 2 && y >= s[t] ) v[t].push_back(y);
else b[t] = 1;
} dfs(1);
go( i, hd[1] ) if ( !b[v] ) g[++m] = s[v];
if ( n < m ) return 0;
sort( h + 1, h + n + 1, cmp ), sort( g + 1, g + m + 1, cmp );
fp( i, 1, m ) if ( h[i] < g[i] ) return 0;
return 1;
}
signed main(){
t_bg = clock();
read(N); int x, y, z; i64 l(1), r(0), mid, ans(-1);
fp( i, 2, N ) read(x), read(y), read(z), addedge( x, y, z ), r += z;
DFS(1), read(M); fp( i, 1, M ) read(w[i]);
while( l <= r ){
mid = ( l + r ) >> 1;
if ( check(mid) ) r = mid - 1, ans = mid;
else l = mid + 1;
} printf( "%lld\n", ans );
t_ed = clock();
fprintf( stderr, "\n========info========\ntime : %.3f\n====================\n", (double)( t_ed - t_bg ) / CLOCKS_PER_SEC );
return 0;
} 
京公网安备 11010502036488号