王室联邦
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 | #include<bits/stdc++.h> using namespace std; const int maxn = 1005; int head[maxn],To[maxn<<2],Next[maxn<<2]; int n,b,cnt,type,top; int Stack[maxn],ans[maxn],belong[maxn]; void add( int u, int v){ Next[++cnt]=head[u]; head[u]=cnt; To[cnt]=v; } void dfs( int u, int fa){ int now=top; for ( int i=head[u];i;i=Next[i]){ int v=To[i]; if (v==fa) continue ; dfs(v,u); if (top-now>=b){ ans[++type]=u; while (top!=now){ belong[Stack[top--]]=type; } } } Stack[++top]=u; } int main(){ scanf ( "%d %d" ,&n,&b); for ( int i=1;i<n;i++){ int u,v; scanf ( "%d %d" ,&u,&v); add(u,v); add(v,u); } dfs(1,0); while (top) belong[Stack[top--]]=type; cout<<type<< "\n" ; for ( int i=1;i<=n;i++){ cout<<belong[i]<< " " ; } cout<< "\n" ; for ( int i=1;i<=type;i++){ cout<<ans[i]<< " " ; } return 0; } |
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】博客园社区专享云产品让利特惠,阿里云新客6.5折上折
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步