AtCoder Regular Contest 117

题目链接

题意：给你一颗n个点的树，让你构造每个点的权值，满足以下条件
1.结点权值a[i]>=1,
2.dis(A,B)<=|a[A]-A[B]|，即结点权值之差大于等于A,B之间的距离
3.结点最大值尽可能小

数据范围：
2≤n≤200000

如果只存在一条链，那么答案即为n。

如果有多条链，那么如果我们处理好了一条链x，那么下一条链y的开头结点的权值一定为x链的最后一个点的权值+y的开头结点与x最后一个结点的长度(即回溯的长度)。我们要让回溯最小即让最长的链最后在处理，这样可以避免最后一个点加上其回溯的长度，导致$\max (a_i)$变大。

跑一遍dfs，找出树的直径有哪些点，我们优先处理不是直径上的点，这样可以使回溯的长度最小化，然后在沿着直径往下找。注意沿着直径走的时候k不用再+1，因为每次k+1均是对于该点后面的点要满足的条件，所以沿着直径走的时候再-1即可。

时间复杂度O(n)

#include<iostream>
#include<cstdio>
#include<vector>
#include<queue>
#include<cstring>
#include<string>                        
#include<algorithm>
#include<fstream>
#include<cmath>
#include<map>
#include<set>
#include<stack>
using namespace std;
typedef long long ll;
const int maxn=1e6+10;    
vector<int>G[maxn];
void add(int x,int y)
{
    G[x].push_back(y);
    G[y].push_back(x);
}
int d[maxn],d1=1,d2=1;     //直径的两个端点
int pre[maxn],vis[maxn];      //pre[i]:i的父亲是谁 vis[i]:i是否在直径上
void dfs(int u,int fa)
{
    d[u]=d[fa]+1;
    pre[u]=fa;
    if(d[u]>d[d1]) {
        d1=u;
    }
    for(int i=0;i<G[u].size();i++)
    {
        int v=G[u][i];
        if(v==fa) continue;
        dfs(v,u);
    }
}
void find(int n)
{
    d1=1; dfs(1,1);                    //找左端点
    d2=d1; d[d1]=0;                    
    dfs(d1,d1);                        //找右端点
    for(int i=d1;i!=d2;i=pre[i])
    vis[i]=1;                              //更新直径上的点
}
int k=0,a[maxn];
void dfs1(int u,int fa)
{
    a[u]=++k;
    if(u==d2) return ;       //遍历到直径的另一个端点
    for(int i=0;i<G[u].size();i++)
    {
        int v=G[u][i];
        if(v==fa||v==pre[u]) continue;
        dfs1(v,u);
    }
    k++;    //因为要在加一段距离，所以k要在加1
    if(vis[u]) {
        k--;
        dfs1(pre[u],u);
    }
}
int main()               
{
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    cout.tie(nullptr);
    int n;cin>>n;
    for(int i=1;i<n;i++)
    {
        int x,y;cin>>x>>y;
        add(x,y);
    }
    find(n);
    dfs1(d1,d1);
    for(int i=1;i<=n;i++)
    cout<<a[i]<<" ";
    return 0;  
}