Educational Codeforces Round 35 F. Tree Destruction 树的直径

题目链接： https://codeforces.com/contest/911/problem/F

题意：

你现在有一棵 $n$ 个结点的树，你现在要进行 $n-1$ 次操作，每次操作你都会选取两个叶子结点 $x$ 和 $y$， 对答案加上 $dis(x,y)$ 后删去两个点中的某一个点，再重复选取操作。 要你求出答案的最大值，以及每次操作的时候选取的点和删除的点。

做法：

其实就感觉是个水题。。可能是因为题目多了来不及做所以难题系数这么高。

假如有一棵树，你要选取两个距离最远的，那么肯定是树的直径。在初始不删除任意点的情况下，一个点肯定是到树直径的某一端的距离为可贡献的最大值。那么如果我们要让总答案最大，肯定是先不动直径上的点，把其他所有的点删光光，然后再处理树的直径上的部分即可。

代码

#include<bits/stdc++.h>
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define rep_e(i,u,v) for(int i=head[u],v=to[i];~i;i=nex[i],v=to[i])
using namespace std;
const int maxn=200005;
const int maxm=400005;
typedef long long ll;
typedef pair<int,int> pii;
int n,link[maxn],fid,dis[maxn],pre[maxn];
int head[maxn],to[maxm],nex[maxm],cnt;
pii mat[maxn];
ll ansl;
vector<int> Link;
vector<pii> ans;
void add(int u,int v){
    to[cnt]=v;nex[cnt]=head[u];
    head[u]=cnt++;
}

void find_far(int u,int f){
    rep_e(i,u,v){
        if(v==f) continue;
        dis[v]=dis[u]+1;
        pre[v]=u;
        if(dis[v]>dis[fid]) fid=v;
        find_far(v,u);
    }
}
void mark(int u,int f,int dis,int aim){
    rep_e(i,u,v){
        if(v==f) continue;
        if(!link[v]){
            if(mat[v].first<dis) mat[v]={dis,aim};
        }
        mark(v,u,dis+1,aim);
    }
}
void dfs(int u,int f){
    //printf("now = %d f = %d\n",u,f);
    rep_e(i,u,v){
        if(v==f||link[v]) continue;
        dfs(v,u);
    }
    if(!link[u]) {
        ans.push_back({u,mat[u].second});
        ansl+=mat[u].first;
    }
}
int main(){
    memset(head,-1,sizeof(head));
    scanf("%d",&n);
    rep(i,2,n){
        int x,y; scanf("%d%d",&x,&y);
        add(x,y); add(y,x);
    }
    find_far(1,-1);
    rep(i,1,n) dis[i]=0,pre[i]=0;
    int onep=fid;
    find_far(fid,-1);
    for(int x=fid;x!=0;x=pre[x]) {
        link[x]=1; Link.push_back(x);
    }
    mark(onep,-1,1,onep); mark(fid,-1,1,fid);
    for(int i=0;i<Link.size();i++){
        dfs(Link[i],-1);
    }
    int sz=Link.size();
    for(int i=0;i<sz-1;i++){
        ans.push_back({Link[i],onep});
        ansl+=sz-i-1;
    }
    printf("%lld\n",ansl);
    for(auto p:ans){
        printf("%d %d %d\n",p.first,p.second,p.first);
    }
    return 0;
}