D52 樹的直徑+貪心 CF911F Tree Destruction

董晓發表於2024-09-20

影片連結:

CF911F Tree Destruction - 洛谷 | 電腦科學教育新生態 (luogu.com.cn)

// 兩次DFS + 貪心 O(n)
#include <iostream>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;

const int N=200005;
int head[N],idx;
struct edge{
  int to,ne;
}e[N<<1];
void add(int x,int y){
  e[++idx]={y,head[x]};head[x]=idx;
}
int n,p,l,r,d[N],fa[N],on[N];
long long ans;
vector<pair<int,int>>s;

void dfs(int x,int f){
  fa[x]=f;
  if(d[x]>d[p]) p=x;
  for(int i=head[x];i;i=e[i].ne){
    int y=e[i].to;
    if(y==f) continue;
    d[y]=d[x]+1;
    dfs(y,x);
  }
}
void work(int x,int rt){
  for(int i=head[x];i;i=e[i].ne){
    int y=e[i].to;
    if(d[y]>d[x]) work(y,on[y]?y:rt);
    //若y在直徑上,則取y
    //若y不在直徑上,則取直徑上的節點
  }
  if(!on[x]){ //若x不在直徑上
    if(d[x]>d[x]+d[r]-d[rt]*2){
      ans+=d[x];
      s.push_back({l,x});
    }
    else{
      ans+=d[x]+d[r]-d[rt]*2;
      s.push_back({r,x});
    }
  }
}
int main(){
  scanf("%d",&n);
  for(int i=1,x,y;i<n;++i){
    scanf("%d%d",&x,&y);
    add(x,y); add(y,x);
  }
  dfs(1,0); l=p; d[p]=0;
  dfs(p,0); r=p;
  for(int i=r;i;i=fa[i]) on[i]=1;
  work(l,l);         //刪外周
  for(;l!=r;r=fa[r]) //刪直徑
    ans+=d[r],s.push_back({l,r});
  printf("%lld\n",ans);
  for(auto i:s)
    printf("%d %d %d\n",i.first,i.second,i.second);
}

相關文章