BZOJ 1086 王室联邦(树分块)

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/qq_37632935/article/details/82150934

题意:给你一棵树，让你把一棵树分成若干块，没块必须是联通的，且大小为b-3b.

思路:树分块的入门操作。我们用建立一个栈每次访问完的节点入栈，继续搜索子树，如果发现当前点的子树中节点数大于b把子树出栈，划为一个块，最后统计一下栈中剩下的节点即可。

#include <vector>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <stdio.h>
using namespace std;

int rot[1005];
int a[1005],stack[1005];
vector<int> q[1005];
int bl[1005];
int b;
int cnt;
int top;
void dfs(int f,int u)
{
	int now=top;
	for(int i=0;i<q[u].size();i++)
	{
		int v=q[u][i];
		if(v==f) continue;
		dfs(u,v);
		if(top-now>=b)
		{
			rot[++cnt]=u;
			while(top!=now)bl[stack[top--]]=cnt;
		}
	}
	stack[++top]=u;
}
int main()
{
	int n;top=0;
	cnt=0;
	scanf("%d%d",&n,&b);
	for(int i=0;i<n-1;i++)
	{
		int x,y;scanf("%d%d",&x,&y);q[x].push_back(y);q[y].push_back(x);
	}
	dfs(1,1);
	while(top) 
	{
		bl[stack[top--]]=cnt;
	}
	printf("%d\n",cnt);
	for(int i=1;i<=n;i++)
	printf("%d ",bl[i]);
	printf("\n");
	for(int i=1;i<=cnt;i++)
	printf("%d ",rot[i]);
	return 0;
}