D - Miracle Tree(思维+构造+树的直径)

https://atcoder.jp/contests/arc117/tasks/arc117_d

---

思路：

比较明显的是树的直径把。

然后构造：先从树的直径开始1，2，3...

如果有分枝，先跑完分枝，然后到叶子了+1

#include<iostream>
#include<vector>
#include<queue>
#include<cstring>
#include<cmath>
#include<map>
#include<set>
#include<cstdio>
#include<algorithm>
#define debug(a) cout<<#a<<"="<<a<<endl;
using namespace std;
const int maxn=2e5+100;
typedef long long LL;
inline LL read(){LL x=0,f=1;char ch=getchar();	while (!isdigit(ch)){if (ch=='-') f=-1;ch=getchar();}while (isdigit(ch)){x=x*10+ch-48;ch=getchar();}
return x*f;}
LL ans[maxn],fat[maxn];
bool used[maxn],vis[maxn];
vector<LL>g[maxn];
LL d=0,maxv=-1;
LL tot=0;
void dfs1(LL u,LL fa,LL dis){
     if(dis>maxv){
        d=u;maxv=dis;
     }
     for(LL i=0;i<g[u].size();i++){
         LL v=g[u][i];
         if(v==fa) continue;
         fat[v]=u;
         dfs1(v,u,dis+1);
     }
}
void dfs(LL u,LL fa){
     ans[u]=++tot;vis[u]=true;
     vector<LL>v1;
     for(LL i=0;i<g[u].size();i++){
         LL v=g[u][i];
         if(v==fa||vis[v]) continue;
         if(!used[v]) dfs(v,u);///先寻找该节点上的不在树的直径上的点
         else v1.push_back(v);
     }
     if(v1.size()){
        for(auto i:v1){
            if(!vis[i]) dfs(i,u);
        }
        return;
     }
     tot++;///非直径上的点的一个叶子访问完了+1
}
int main(void){
   cin.tie(0);std::ios::sync_with_stdio(false);
   LL n;cin>>n;
   for(LL i=1;i<n;i++){
       LL u,v;cin>>u>>v;
       g[u].push_back(v);
       g[v].push_back(u);
   }
   dfs1(1,0,0);
   LL st=d;
   maxv=-1;
   fat[st]=0;
   dfs1(st,0,0);
   for(LL i=d;i!=0;i=fat[i]){
      used[i]=true;
   }
   dfs(d,0);
   for(LL i=1;i<=n;i++){
       cout<<ans[i]<<" ";
   }
   cout<<"\n";
   return 0;
}