一个比较有趣的树形 DP,情况比较多。
【题目简述】
给定一棵树,求三条相连的边,其边权之和最大。
【思路】
以 X 代表当前节点,S 表示儿子,G 表示孙子,P 表示父节点。
首先把树建出来,在以下图中,我们模拟二号点的 DP 过程,考虑以下几种情况:
-
有一条边指向父节点时
-
FG(Father Grandson):一条边指向父节点,另外两条边指向孙子。
需要维护指向孙子的路径的最大值。
-
FSS(Father Son Son):一条边指向父节点,另外两条边指向两个不同的儿子。
需要维护指向儿子的路径的最大、次大值。
-
-
没有边指向父节点时
-
GS(Grandson Son):一条边指向孙子,另外一条边指向另一个儿子。
需要维护指向儿子的路径的最大、次大值和指向孙子的路径的最大、次大值(可能这两条路径重合)
-
SSS(Son Son Son):三条边指向三个不同的儿子。
需要维护指向儿子的路径的最大、次大、第三大值。
-
-
要维护的东西:指向儿子的路径的最大、次大、第三大值,指向孙子的路径的最大、次大值。
于是代码就很好写了。
【Code】
#include <bits/stdc++.h>
#define int long long
using namespace std;vector<pair<int,int>>Edge[200005];
int n,u,w,ans;int DFS(int u,int fa,int W){int max11=0,flag11=0,max12=0,flag12=0,max13=0,flag13=0; //max1x,flag1x:向 Dis(X,S) x 大值及儿子编号 int max21=0,flag21=0,max22=0,flag22=0; //max2x,flag2x:向 Dis(X,G) 的路径的 x 大值及儿子编号 for(auto it:Edge[u]){int v=it.first,DisXS=it.second;if(v!=fa){int DisSG=DFS(v,u,DisXS); //Dis(S,G) 的最小值//如果儿子还有儿子(有孙子),更新到 Dis(X,G) 的最大、次大值 if(DisSG!=0){int DisXG=DisXS+DisSG;if(max21<DisXG) max21=max21,flag22=flag21,max21=DisXG,flag21=v;else if(max22<DisXG) max22=DisXG,flag22=v;}//更新到 Dis(X,S) 的最大、次大、第三大值 if(max11<DisXS) max13=max12,flag13=flag12,max12=max11,flag12=flag11,max11=DisXS,flag11=v;else if(max12<DisXS) max13=max12,flag13=flag12,max12=DisXS,flag12=v;else if(max13<DisXS) max13=DisXS,flag13=v;}}//W 表示到父节点的路径长度 if(u!=1) ans=max(ans,max21+W); //FGif(u!=1) ans=max(ans,max11+max12+W); //FSSif(flag11!=flag21) ans=max(ans,max11+max21); //GS(两条路径不重合) if(flag11==flag21) ans=max({ans,max11+max22,max12+max21}); //(两条路径重合,其中一条选择次长路) ans=max(ans,max11+max12+max13); //SSSreturn max11;
}signed main()
{scanf("%lld",&n);for(int v=2;v<=n;v++){scanf("%lld%lld",&u,&w);Edge[u].push_back({v,w});Edge[v].push_back({u,w});}DFS(1,0,0);printf("%lld",ans);return 0;
}
【闲话】
祝所有看到这篇题解的人 CSP2024 RP++!