CF 1805 D. A Wide, Wide Graph (*1800) 思维 + 树的直径

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;
}