洛谷 P2325 王室联邦

原题链接：https://www.luogu.com.cn/problem/P2325
树上分块思想，首先只考虑一棵以u为根节点高度为2的子树（从下往上推），如果说它的子结点个数大于b，那么可以将它的子节点分为若干块，每一块都等于b并且认为他们的省会为根节点。最后会有多余的一块，子结点个数小于b，此时可以将它与根节点合成一块，这一块就可以与它相邻的其他子树多余的一块合并，可画图理解。
过程使用一个栈来存结点，遍历一棵子树时，记录此时的栈大小，在将子树中的节点加入栈中，当加入的元素大于等于b就可以分成一块。（大于b的情况就是子树中两颗（或多颗）小子树的合并）。
code:

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> pi;
typedef complex <double> cp;
#define debug(a) cout<<#a<<":"<<a<<endl;
#define fr freopen("in.txt","r",stdin);
#define Fill(x,a) memset(x,a,sizeof(x))
#define cpy(a,b) memcpy(a,b,sizeof(a))
const double eps = 1e-8;
const double inf = 1e20;
const double PI = acos(-1);
const int INF=0x3f3f3f3f;
const int N=1e6+7;
const int mod=1e9+7;
int maxn,minn;
int T,n,m,q;
struct edges{
	int u,next;
}edge[N];
int cnt,top;
int head[N];
int belong[N];
int root[N];
int sta[N];

void init(){
	cnt = 1;
	Fill(head,0);
	return ;
}

void add(int u,int v){
	edge[cnt].u = v;
	edge[cnt].next = head[u];
	head[u] = cnt ++ ;
	return ;
}

void dfs(int u,int p){
	int pos,v;
	pos = top;
	for(int i = head[u]; i != 0; i = edge[i].next){
		v = edge[i].u;
		if(v == p)	continue;
		dfs(v,u);
		if(top - pos >= m){
			root[++cnt] = u;
			for(;top != pos; top -- ){
				belong[sta[top]] = cnt;
			}
		}
	}
	sta[++top] = u;
	return ;
}

int main(){
	int u,v;
	cin >> n >> m;
	init();
	for(int i = 1; i <= n - 1; i ++ ){
		scanf("%d%d",&u,&v);
		add(u,v);
		add(v,u);
	}
	if(u < m){
		puts("0");
	}
	else{
		top = cnt = 0;
		dfs(1,0);
		for(int i = 1; i <= top; i ++ ){
			belong[sta[i]] = cnt;
		}
		printf("%d\n",cnt);
		for(int i = 1; i <= n; i ++ ){
			printf("%d ",belong[i]);
		}
		printf("\n");
		for(int i = 1; i <= cnt; i ++ ){
			printf("%d ",root[i]);
		}
		printf("\n");
	}
	return 0;
}