树分块——BZOJ1086/Luogu2325 [SCOI2005]王室联邦

题面：BZOJ1086 Luogu2325
树分块模板题？
很懵逼我这个算不算树分块QAQ，暂且当它是吧到时候学树上莫队的时候再说
因为题目要求随便一组即可，所以我们可以考虑一下贪心
当子树大小超过B的时候单独作为一个省，子树根为省会（没算在那个省里），然后清空一下当前的子树大小
然后最后会剩下这么点城市，这些城市大小必定 < B（显然，否则也被分进另外一个省了）
于是把这一小部分分进相邻的随便一个省就好了，这样那个省大小也不会超过3B的（为什么？因为一开始分的时候大小肯定小于2B否则也分两个省了，再加上B不会超过3B）
然后输出就行了

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<iostream>
#include<cstdlib>
#include<set>
#include<map>
#include<string>
#include<ctime>
#include<queue>
#include<vector>
using namespace std;
int q[100001];
int n,B,s[100001],ans,an[100001],sh[100001];
int nedge=0,p[100001],nex[100001],head[100001];
inline void addedge(int a,int b){
    p[++nedge]=b;nex[nedge]=head[a];head[a]=nedge;
}
inline void dfs(int x,int fa){
    q[++q[0]]=x;
    for(int k=head[x];k;k=nex[k])if(p[k]!=fa){
        dfs(p[k],x);
        if(s[x]+s[p[k]]>=B){
            s[x]=0;sh[++ans]=x;
            while(q[q[0]]!=x)an[q[q[0]--]]=ans;
        }else s[x]+=s[p[k]];
    }
    s[x]++;
}
inline void dfss(int x,int fa,int c){
    if(an[x])c=an[x];else an[x]=c;
    for(int k=head[x];k;k=nex[k])if(p[k]!=fa)dfss(p[k],x,c);
}
int main()
{
    scanf("%d%d",&n,&B);
    if(n<B)return puts("0")&0;
    for(int i=1;i<n;i++){
        int x,y;scanf("%d%d",&x,&y);
        addedge(x,y);addedge(y,x);
    }
    dfs(1,0);
    if(!ans)ans++;dfss(1,0,ans);
    printf("%d\n",ans);for(int i=1;i<=n;i++)printf("%d ",an[i]);puts("");
    for(int i=1;i<=ans;i++)printf("%d ",sh[i]);
    return 0;
}