思路
假设源点已经确定为1,那么直接跑树形DP即可.表示
到叶子节点的最大流量(不考虑父亲节点的限制,也就是说看成父亲流到
的流量为无穷),
但是现在源点不一定为1,考虑怎么快速地"换根".设为
为根时真正的答案.首先,它流到以
为根时就是其儿子节点的流量是
(注意
是叶子节点时要赋为0).流到以
为根时为
父亲,而以
为根时是其儿子节点的最大流量为
.如果不减去
就会导致多出计算
流到
的流量.然后就可以
转移了.总复杂度为
.
代码
#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 = 2e5 + 5;
int T, N;
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 f[MAXN], g[MAXN]; bool b[MAXN];
void DFS1( int u, int fa ){
b[u] = 1, f[u] = 0;
go( i, hd[u] ) if ( v != fa ){
DFS1(v, u), b[u] = 0;
f[u] += min( f[v], val[i] );
} if ( b[u] ) f[u] = INT_MAX;
}
void DFS2( int u, int fa, int vl ){
g[u] = b[u] ? 0 : f[u], g[u] += vl;
go( i, hd[u] ) if ( v != fa ) DFS2(v, u, min(g[u] - min(val[i], f[v]), val[i]));
}
signed main(){
t_bg = clock();
read(T);
while( T-- ){
read(N); int x, y, z; memset( hd, 0, sizeof hd ), tot = 0;
fp( i, 2, N ) read(x), read(y), read(z), addedge( x, y, z );
DFS1(1, 0), DFS2(1, 0, 0); int ans(0);
fp( i, 1, N ) cmax( ans, g[i] );
printf( "%d\n", ans );
}
t_ed = clock();
fprintf( stderr, "\n========info========\ntime : %.3f\n====================\n", (double)( t_ed - t_bg ) / CLOCKS_PER_SEC );
return 0;
} 
京公网安备 11010502036488号