标签：ch min int son DP go 节点 换根 287
比较好想的换根 DP，但是有坑点。
我的做法需要的代码行数应该比较少。。去掉头部就 40 行左右。
分析
约定 \(f[u]\) 表示以 \(u\) 为根节点的时候的最大流量，\(w(u, v)\) 为 \(u, v\) 之间的边权。
首先考虑根节点为 \(u\) 的时候如何统计 \(f[u]\)（也就是统计子树 \(u\) 的结果）：
- \(u\) 的子节点 \(son[u]\) 能够提供的流量为 \(f[son[u]]\)。
- 因此，相应能够为 \(f[u]\) 提供的贡献为 \(\min(f[son[u]], w(u, son[u]))\)
这样，以 \(1\) 为根节点，我们就能够通过一次 \(dfs\) 将 \(f[1]\) 的结果处理出来。
需要注意的是，为了保证源点的流量是无穷，我们应该加一个特判：当遇到源点（叶节点）时，首先将其 \(f\) 值设为 \(\inf\)，最后在 \(dfs\) 后为了保证 \(f\) 值的正确性我们再将其设为 \(0\)。
下面考虑换根，将所有节点为根的时候的结果再次使用一次 \(dfs\) 统计出来。
那么转移的过程是：
- 设现在已经处理好了 \(f[u]\)。
- 考虑使用父节点 \(u\) 的 \(f\) 值更新 \(f[son[u]]\)，换句话说就是考虑 \(u\) 能为 \(son[u]\) 提供的流量是多少。
- 那么，\(f[son[u]]\) 需要加上的是 \(u\) 点减去 \(son[u]\) 所在的子树的贡献 \(\min(f[son[u]], w(u, son[u])\)，也就是：\(f[son[u]] += f[u] - \min(f[son[u]], w(u, son[u]))\)。
但是这里有一个问题：如果父节点 \(u\) 可以作为源点（也就是其度数为 \(1\)），会使 \(u\) 为 \(son[u]\) 提供的流量为 \(\min(w(u, son[u]), \inf) = w(u, son[u])\)。但这个情况其实只会出现于：\(u\) 为根节点 \(1\) 且 \(1\) 的度数为 \(1\) 的情况，对此特判就好了。
// Problem: 积蓄程度
// Contest: AcWing
// URL: https://www.acwing.com/problem/content/289/
// Memory Limit: 64 MB
// Time Limit: 1000 ms
//
// Powered by CP Editor (https://cpeditor.org)
#include<bits/stdc++.h>
using namespace std;
#define debug(x) cerr << #x << ": " << (x) << endl
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define dwn(i,a,b) for(int i=(a);i>=(b);i--)
#define pb push_back
#define all(x) (x).begin(), (x).end()
#define x first
#define y second
using pii = pair<int, int>;
using ll = long long;
inline void read(int &x){
int s=0; x=1;
char ch=getchar();
while(ch<'0' || ch>'9') {if(ch=='-')x=-1;ch=getchar();}
while(ch>='0' && ch<='9') s=(s<<3)+(s<<1)+ch-'0',ch=getchar();
x*=s;
}
const int N=4e5+5, INF=0x3f3f3f3f;
vector<pii> g[N];
int f[N];
int n;
void dfs1(int u, int fa){
f[u]=(g[u].size()==1 && u!=1? INF: 0);
for(auto [go, w]: g[u]){
if(go==fa) continue;
dfs1(go, u);
f[u]+=min(w, f[go]);
}
}
void dfs2(int u, int fa){
for(auto [go, w]: g[u]){
if(go==fa) continue;
f[go]+=min(w, (u==1 && g[u].size()==1? INF: f[u])-min(w, f[go]));
dfs2(go, u);
}
}
int main(){
int T; cin>>T;
while(T--){
cin>>n;
rep(i,1,n-1){
int u, v, w; read(u), read(v), read(w);
g[u].pb({v, w}), g[v].pb({u, w});
}
dfs1(1, 0);
rep(i,1,n) if(f[i]==INF) f[i]=0;
dfs2(1, 0);
int res=0;
rep(i,1,n) res=max(res, f[i]);
cout<<res<<endl;
rep(i,1,n<<1) g[i].clear();
}
return 0;
}
标签：ch,min,int,son,DP,go,节点,换根,287 来源： https://www.cnblogs.com/Tenshi/p/15954483.html
本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享； 2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关； 3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关； 4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除； 5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。