D - Miracle Tree 树的直径+思维
题意:给你一颗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 ) \max(a_{i}) max(ai)变大。
跑一遍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;
}