BZOJ 1086 & 类树的分块

题意:

　　“余”人国的国王想重新编制他的 国家。他想把他的国家划分成若干个省，每个省都由他们王室联邦的一个成员来管理。他的国家有n个城市，编号为1..n。一些城市之间有道路相连，任意两个 不同的城市之间有且仅有一条直接或间接的道路。为了防止管理太过分散，每个省至少要有B个城市，为了能有效的管理，每个省最多只有3B个城市。每个省必须 有一个省会，这个省会可以位于省内，也可以在该省外。但是该省的任意一个城市到达省会所经过的道路上的城市（除了最后一个城市，即该省省会）都必须属于该 省。一个城市可以作为多个省的省会。聪明的你快帮帮这个国王吧！

sol:

　　　膜拜popoqqq大神犇...题解非常直观清晰...附个链接好了...http://blog.csdn.net/popoqqq/article/details/42772237

code:

　　

/**************************************************************
    Problem: 1086
    User: YCuangWhen
    Language: C++
    Result: Accepted
    Time:12 ms
    Memory:3616 kb
****************************************************************/

/*==========================================================================
# Last modified: 2016-02-29 14:01
# Filename: 1086.cpp
# Description:树的分块
==========================================================================*/
#define me AcrossTheSky
#include <cstdio>
#include <cmath>
#include <ctime>
#include <string>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm> 

#include <set>
#include <map>
#include <stack>
#include <queue>
#include <vector> 

#define lowbit(x) (x)&(-x)
#define FOR(i,a,b) for((i)=(a);(i)<=(b);(i)++)
#define FORP(i,a,b) for(int i=(a);i<=(b);i++)
#define FORM(i,a,b) for(int i=(a);i>=(b);i--)
#define ls(a,b) (((a)+(b)) << 1)
#define rs(a,b) (((a)+(b)) >> 1)
#define getlc(a) ch[(a)][0]
#define getrc(a) ch[(a)][1] 

#define maxn 100000
#define maxm 100000
#define pi 3.1415926535898
#define _e 2.718281828459
#define INF 1070000000
using namespace std;
typedef long long ll;
typedef unsigned long long ull; 

template<class T> inline
void read(T& num) {
    bool start=false,neg=false;
    char c;
    num=0;
    while((c=getchar())!=EOF) {
        if(c=='-') start=neg=true;
        else if(c>='0' && c<='9') {
            start=true;
            num=num*10+c-'0';
        } else if(start) break;
    }
    if(neg) num=-num;
}
/*==================split line==================*/
int top=0,sume=0;
int n,b,cnt=0;
int to[maxm],belong[maxn],next[maxm],first[maxn];
int s[maxm],root[maxn];
void addedge(int x,int y){
    sume++; to[sume]=y; next[sume]=first[x]; first[x]=sume;
}
void dfs(int node,int fa){
    int bottom=top;
    for (int i=first[node];i!=-1;i=next[i])
    if (to[i]!=fa){
        dfs(to[i],node);
        if (top-bottom>=b){
            root[++cnt]=node;
            while (top!=bottom) belong[s[top--]]=cnt;
        }
    }
    s[++top]=node;
}
int main(){
    read(n); read(b);
    FORP(i,1,n) first[i]=-1;
    FORP(i,1,n-1) {
        int x,y; read(x); read(y);
        addedge(x,y); addedge(y,x);
    }
    dfs(1,0);
    while (top) belong[s[top--]]=cnt;
    printf("%d\n",cnt);
    FORP(i,1,n) printf("%d ",belong[i]); printf("\n");
    FORP(i,1,cnt) printf("%d ",root[i]);
}