CF 1805 D. A Wide, Wide Graph (*1800) 思维 + 树的直径
题意:
思路:
若当前点到最远的点的距离 \(< k\) , 说明 \(x\) 自己成为一个联通块。
并且我们知道距离任意一点最远的点一定是树直径的一个端点。
反之,则与直径端点在同一个联通块。
所以一个点要么独立成为联通块,要么和直径端点在一个联通块。
\(dfs\) 求出直径两个端点,并且维护每个点到端点的距离即可。
代码:
#include<bits/stdc++.h>
using namespace std;
using i64=long long;
void Showball(){
int n;
cin>>n;
vector<vector<int>> e(n);
for(int i=1;i<n;i++){
int u,v;
cin>>u>>v;
u--;
v--;
e[u].push_back(v);
e[v].push_back(u);
}
vector<int> st(n),dis(n);
function<void(int)> dfs=[&](int u){
for(auto v:e[u]){
if(st[v]) continue;
st[v]=true;
dis[v]=dis[u]+1;
dfs(v);
}
};
st[0]=1;
dfs(0);
int maxn=0,S,E;
for(int i=0;i<n;i++){
if(dis[i]>maxn){
maxn=dis[i];
S=i;
}
dis[i]=st[i]=0;
}
st[S]=1;
dfs(S);
auto dis2=dis;
maxn=0;
for(int i=0;i<n;i++){
if(dis[i]>maxn){
maxn=dis[i];
E=i;
}
dis[i]=st[i]=0;
}
st[E]=1;
dfs(E);
vector<int> ans(n+1);
for(int i=0;i<n;i++){
if(i!=E) ans[max(dis[i],dis2[i])+1]++;
}
ans[0]=1;
for(int i=1;i<=n;i++) ans[i]+=ans[i-1];
for(int i=1;i<=n;i++) cout<<ans[i]<<" \n"[i==n];
}
int main(){
ios::sync_with_stdio(false);
cin.tie(nullptr);
int t=1;
//cin>>t;
while(t--){
Showball();
}
return 0;
}
标签:Wide,1805,1800,int,dfs,st,端点,dis
From: https://www.cnblogs.com/showball/p/18448187