題目鏈接

Solution

1.先找出樹的直徑.
2.遍歷直徑沿途的每一個節點以及它的子樹.
3.然后對于每個非直徑節點直接統計答案,令直徑的兩個端點為 \(x_1,x_2\) .
\[Ans=\sum{Max(dis(i,x1),dis(i,x2))}\]
最后再單獨把直徑拎出來,單獨統計一次就好了.
正確性證明:
redbag 一句話解決:

如果說這其中的某個答案不是最優,那么找的直徑不就是錯的么。

"跑的飛快。"

Code

#include<bits/stdc++.h> using namespace std; const int maxn=200008; struct sj{int to,next; }a[maxn*2]; int head[maxn],size,x,y,road[maxn]; int root,flag,n,cnt,far[5],kk,vis[maxn]; int dis[maxn][3],v[maxn],num; long long ans; int ans1[maxn],ans2[maxn],siz; void add(int x,int y) {a[++size].to=y;a[size].next=head[x];head[x]=size; }void dfs(int x) {v[x]=1; if(cnt>kk)far[root==1?1:2]=x,kk=cnt; for(int i=head[x];i;i=a[i].next){int tt=a[i].to;if(!v[tt])cnt++,dfs(tt);} v[x]=0; cnt--; }void fuck(int x) {v[x]=1;if(x==far[2]){flag=1;}for(int i=head[x];i;i=a[i].next){int tt=a[i].to;if(!v[tt])fuck(tt);if(flag){road[++num]=x;vis[x]=1;break;}}v[x]=0; }void getans(int x) {v[x]=1;for(int i=head[x];i;i=a[i].next){int tt=a[i].to;if(!v[tt]&&!vis[tt])getans(tt),ans+=max(dis[tt][1],dis[tt][2]);}if(dis[x][1]>dis[x][2])ans1[++siz]=far[1];else ans1[++siz]=far[2];ans2[siz]=x;v[x]=0; }void getdis(int x,int to) {v[x]=1;for(int i=head[x];i;i=a[i].next){int tt=a[i].to;if(!v[tt]){dis[tt][to]=dis[x][to]+1; getdis(tt,to);}}v[x]=0; }int main() {cin>>n;for(int i=1;i<n;i++){scanf("%d%d",&x,&y),add(x,y), add(y,x);}root=1; dfs(root); root=far[1]; kk=0; cnt=0; dfs(root); fuck(root);getdis(far[1],1); getdis(far[2],2);for(int i=2;i<num;i++)getans(road[i]),siz--;for(int i=1;i<=kk;i++)ans+=i;cout<<ans<<endl;for(int i=1;i<=siz;i++)cout<<ans1[i]<<' '<<ans2[i]<<' '<<ans2[i]<<endl;for(int i=1;i<num;i++)cout<<far[1]<<' '<<road[i]<<' '<<road[i]<<endl; }

轉載于:https://www.cnblogs.com/Kv-Stalin/p/9371937.html

《新程序員》：云原生和全面數字化實踐50位技術專家共同創作，文字、視頻、音頻交互閱讀

總結

以上是生活随笔為你收集整理的CF911F Tree Destruction (树的直径,贪心)的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。