BZOJ 1086 SCOI2005 王室联邦 块状树

朴素的分块方式是贪心 能加就加 这种方法存在着严重的效率问题 可以被菊花卡成O(n)块

因此我们可以为其它的块预留位置 如果一块大小刚好>=b 就将这坨东西分成一块

首先任选一点开始深搜 维护一个栈 每个点退出递归时压栈 自下至上进行合并

如果某棵子树深搜完之后栈内元素数>=b 就把当前的栈内元素合并为一个块

但是这种方法存在一个问题 就是如果某棵子树深搜之后不到b 去深搜下一个子树 可能在下一个子树内部的某个位置超过b

这样会导致分成的块不连通

因此我们在每次进入递归时维护一个栈底，对于当前子树来说这个栈底就是整个栈的底，栈底以下的元素不能修改或弹栈

这样当一棵子树深搜过后由于子树内未分块节点不超过b，之前搜过的未分块节点数也不超过b，因此每块不超过2b

那么题目为什么给了3b呢？

深搜结束后可能会剩余一些节点，这些节点的数量不超过b，而且一定与当前分出的最后一块连通

因此我们将剩余节点分到最后一块中，可以保证最后一块的大小不超过3b

一遍深搜即可出解。

#include <iostream>
#include <cstring>
#include <vector>
#include <algorithm>
#include <cstdio>
#include <stack>
using namespace std;
#define maxn 2100
#define pb(a) push_back(a)
int n,m,belong[maxn];
vector<int>g[maxn];
int top,num,sta[maxn],root[maxn];
void dfs(int now,int from)
{
    int v,bot=top;
    for(int i=0;i<g[now].size();i++)
    {
        v=g[now][i];
        if(v==from) continue;
        dfs(v,now);
        if(top-bot>=m)
        {
            num++;
            root[num]=now;
            while(top>bot)
            {
                belong[sta[top]]=num;
                top--;
            }
        }
    }
    top++;
    sta[top]=now;
}
int main()
{
    int a,b;
    scanf("%d%d",&n,&m);
    for(int i=1;i<n;i++)
    {
        scanf("%d%d",&a,&b);
        g[a].pb(b);
        g[b].pb(a);
    }
    dfs(1,0);
    for(;top;top--)  belong[sta[top]]=num;
    printf("%d\n",num);
    for(int i=1;i<=n;i++)
        printf("%d ",belong[i]);
    printf("\n");
    for(int i=1;i<=num;i++)
        printf("%d ",root[i]);
    printf("\n");
    return 0;
}