bzoj1086 [SCOI2005]王室联邦（树分块）

题目链接

分析：
很有意思的一道题
最开始看这道题，感觉不难，就YY了一种做法：
dfs整棵树，如果某子树的大小满足条件就划分成一个“省”，省会就是子树的根
如果有剩下的点，那么这些点一定集中在根的部分
我们视情况把ta们分成新的一个省份，或者并到最后分出的一个省份中

写完了当然是WA啦
后来才考虑到，如果有一棵树只有两层，儿子非常多，这样把子树整体考虑可能就不行了

正解：
我们要从分利用这个条件：省会不一定在省内
开一个栈，dfs整棵树
如果结点x的几棵子树的size加起来符合要求，
就把这几棵子树划分成一个省，省会就是当前结点x
至于那些不能到达要求的子树，我们就把ta们的size统计到x上，回溯到上一层

最后我们可能会剩下几个点
这样我们划分出来的省会大小一定不会超过2*B
最后剩下的结点一定不超过B（不然就可以在回溯的过程中再划分出一个省份）
我们直接把剩下的点并到最后划分出来的省份中

#include<cstdio>
#include<cstring>
#include<iostream>
#include<cstring>

using namespace std;

const int N=1005;
struct node{
    int x,y,nxt;
};
node way[N<<1];
int n,B,st[N],tot=0;
int S[N],tt=0,ans[N],wh[N],cnt=0,size[N],sz[N];
bool ff=0;

void add(int u,int w)
{
    tot++;
    way[tot].x=u; way[tot].y=w; way[tot].nxt=st[u];st[u]=tot;
    tot++;
    way[tot].x=w; way[tot].y=u; way[tot].nxt=st[w];st[w]=tot;
}

void solve(int now,int fa)
{
    S[++tt]=now;                            //栈 
    for (int i=st[now];i;i=way[i].nxt)
        if (way[i].y!=fa)
        {
            solve(way[i].y,now);
            if (size[now]+size[way[i].y]>=B) {
                sz[++cnt]=size[now]; size[now]=0;
                ans[cnt]=now;
                while (S[tt]!=now)
                    wh[S[tt--]]=cnt;
            }
            else size[now]+=size[way[i].y];
        }
    size[now]++;                          //加上根结点,向上回溯 
}

int main()
{
    scanf("%d%d",&n,&B);
    for (int i=1;i<n;i++)
    {
        int u,w;
        scanf("%d%d",&u,&w);
        add(u,w);
    }
    if (B>n) {
        printf("0");
        return 0;
    } 

    solve(1,0);
    for (int i=cnt;i>=1;i--)
        if (tt+sz[i]<=3*B)
        {
            while (tt) wh[S[tt--]]=i;
            break;
        } 

    printf("%d\n",cnt);
    for (int i=1;i<=n;i++) printf("%d ",wh[i]);
    puts("");
    for (int i=1;i<=cnt;i++) printf("%d ",ans[i]);

    return 0;
}