题意:给一颗树,删掉一条边然后连上一条边,求改变后的树的最大直径。
题解:树形dp记录每个节点向下的最长路径,次长路径,次次长路径,还有以每个节点为根时候的树的直径,通过树形dp计算出删除父节点与子节点之间边后的两个树的直径,然后加起来取最大值就是答案。
AC代码:
#include<stdio.h> #include<vector> #include<iostream> using namespace std; vector<int>vt[500005]; int dp[500005][3],ans; int zhi[500005]; void dfs(int u,int fa) { for(int i=0;i<vt[u].size();i++) { int to=vt[u][i]; if(to==fa)continue; dfs(to,u); if(dp[to][0]+1>=dp[u][0])dp[u][2]=dp[u][1],dp[u][1]=dp[u][0],dp[u][0]=dp[to][0]+1; else if(dp[to][0]+1>=dp[u][1])dp[u][2]=dp[u][1],dp[u][1]=dp[to][0]+1; else if(dp[to][0]+1>=dp[u][2])dp[u][2]=dp[to][0]+1; zhi[u]=max(zhi[u],zhi[to]); } zhi[u]=max(zhi[u],dp[u][0]+dp[u][1]); } void dfs(int u,int fa,int sum,int zj) { int b[2]={0}; sum++; for(int i=0;i<vt[u].size();i++) { int to=vt[u][i]; if(to==fa)continue; if(zhi[to]>=b[0])b[1]=b[0],b[0]=zhi[to]; else if(zhi[to]>=b[1])b[1]=zhi[to]; } for(int i=0;i<vt[u].size();i++) { int to=vt[u][i]; if(to==fa)continue; if(dp[to][0]+1==dp[u][0]) { if(b[0]==zhi[to])ans=max(ans,max(b[1],max(zj,max(sum+dp[u][1],dp[u][1]+dp[u][2])))+zhi[to]+1); else ans=max(ans,max(b[0],max(zj,max(sum+dp[u][1],dp[u][1]+dp[u][2])))+zhi[to]+1); dfs(to,u,max(sum,dp[u][1]),max(zj,dp[to][1]+dp[to][2])); } else if(dp[to][0]+1==dp[u][1]) { if(b[0]==zhi[to])ans=max(ans,max(b[1],max(zj,max(sum+dp[u][0],dp[u][0]+dp[u][2])))+zhi[to]+1); else ans=max(ans,max(b[0],max(zj,max(sum+dp[u][0],dp[u][0]+dp[u][2])))+zhi[to]+1); dfs(to,u,max(sum,dp[u][0]),max(zj,dp[to][0]+dp[to][2])); } else { if(b[0]==zhi[to])ans=max(ans,max(b[1],max(zj,max(sum+dp[u][0],dp[u][0]+dp[u][1])))+zhi[to]+1); else ans=max(ans,max(b[0],max(zj,max(sum+dp[u][0],dp[u][0]+dp[u][1])))+zhi[to]+1); dfs(to,u,max(sum,dp[u][0]),max(zj,dp[to][0]+dp[to][1])); } } } int main() { int n; scanf("%d",&n); for(int i=0;i<n-1;i++) { int u,v; scanf("%d%d",&u,&v); vt[u].push_back(v); vt[v].push_back(u); } dfs(1,1); dfs(1,1,-1,0); printf("%d\n",ans); }