POJ 3585 :Accumulation Degree (树形DP + 两次扫描换根法)

在这里插入图片描述
题目大意:给你一棵树,每条边有一个最大容量,根结点可以流出无限的流,叶子结点可以接收无限多的流,问你以哪一个点为原点整棵树的总容量最大。

其实就是在问以那个点为原点最大流最大,数据很大肯定不是用最大流算法来跑 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;
}
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值