题目大意:给你一棵树,每条边有一个最大容量,根结点可以流出无限的流,叶子结点可以接收无限多的流,问你以哪一个点为原点整棵树的总容量最大。
其实就是在问以那个点为原点最大流最大,数据很大肯定不是用最大流算法来跑 n 遍,因为整体结构是一棵树,容易想到两次DFS处理出每一个点作为根节点的最大流,再找答案。
(所谓两次扫描换根法就是两次DFS扫描得到每一个点作为根结点的解,了解之后发现就是自己用过的东西,第二次DFS时注意只有一条边的根结点,因为要计算子结点的贡献,子节点的贡献 要加上 min(父节点不包含该子节点时的贡献,边容量),但是只有一条边的根结点的贡献全部来自这个子结点,扣去后就得到0,但实际上要加上不是0而是 w (连接根结点的边),因为这时候子结点作为新的根节点可以有无限的多的流)
(else 部分为什么不用分类讨论?如果 v 是原来树的叶子结点,它对u的贡献应该就是 w 而不是 min(sum[v] ,w),这里min(sum[v],w) 是等于0的因为sum[v] = 0。原因是,因为v 对 u的贡献是 w,所以sum[u]是绝对大于 w的,而sum[v] 绝对不会大于 w,因此 sum[v] 不可能作为最终答案,没有分类讨论的结果是导致 sum[v] 偏大了,但不影响最后答案)
正确的写法应该是:
(而我由于只考虑了else里的分类情况而没有考虑 if(d[u] == 1) 的情况,WA到死)
参考样例:
1
4
1 2 3
1 3 1
1 4 1
最终答案是5,但如果输出sum[2],会发现对于第一种写法 输出的是3,第二种写法输出的是 2,正确结果显然是2。
#include<iostream>
#include<stdio.h>
#include<string.h>
using namespace std;
const int maxn = 2e5 + 100;
typedef long long ll;
struct ss{
int v,w,nxt;
}edg[maxn * 2];
int head[maxn],cnt;
int t,n;
void add(int u,int v,int w) {
edg[cnt].v = v;
edg[cnt].nxt = head[u];
edg[cnt].w = w;
head[u] = cnt++;
}
int sum[maxn],d[maxn];
void dfs(int u,int fa) {
sum[u] = 0;
for(int i = head[u]; i + 1; i = edg[i].nxt) {
int v = edg[i].v, w = edg[i].w;
if(v == fa) continue;
dfs(v,u);
if(d[v] == 1) sum[u] += w;
else sum[u] += min(sum[v],w);
}
}
void dfs2(int u,int fa) {
for(int i = head[u]; i + 1; i = edg[i].nxt) {
int v = edg[i].v, w = edg[i].w;
if(v == fa) continue;
if(d[u] == 1)
sum[v] += w;
else {
if(d[v] == 1) sum[v] += min(sum[u] - w,w);
else sum[v] += min(sum[u] - min(sum[v],w),w);
}
dfs2(v,u);
}
}
int main() {
scanf("%d",&t);
while(t--) {
scanf("%d",&n);
cnt = 0;
for(int i = 1; i <= n; i++) d[i] = 0,head[i] = -1;
for(int i = 1; i < n; i++) {
int u,v,w;
scanf("%d%d%d",&u,&v,&w);
d[u]++;d[v]++;
add(u,v,w);
add(v,u,w);
}
dfs(1,-1);
dfs2(1,-1);
int res = 0;
for(int i = 1; i <= n; i++)
res = max(res,sum[i]);
printf("%d\n",res);
}
return 0;
}