bzoj1086: [SCOI2005]王室聯邦（貪心）

題目傳送門
。

解法：
子樹大於等於B就直接那啥嘛。
其實不用管3B的因為你遇到B就分了。
然後剩下的也肯定不大於B，那麼加起來也肯定不超過3B。
用一個棧記錄下訪問過的點就好了。
一開始寫錯了還掛了一次。
因為棧裡有可能有別的省的點啊。

程式碼實現：

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
using namespace std;
struct node {int x,y,next;}a[2100];int len,last[1100];
void ins(int x,int y) {len++;a[len].x=x;a[len].y=y;a[len].next=last[x];last[x]=len;}
int tot[1100],sta[1100],top,belong[1100],B,cnt,ans[1100];
void dfs(int x,int fa) {
    sta[++top]=x;
    for(int k=last[x];k;k=a[k].next) {
        int y=a[k].y;
        if(y!=fa) {
            dfs(y,x);tot[x]+=tot[y];
            if(tot[x]>=B) {
                tot[x]=0;cnt++;ans[cnt]=x;
                while(sta[top]!=x)belong[sta[top--]]=cnt;
            }
        }
    }tot[x]++;
}
void find(int x,int fa,int s) {
    if(belong[x]!=0)s=belong[x];
    else belong[x]=s;
    for(int k=last[x];k;k=a[k].next) {
        int y=a[k].y;
        if(y!=fa) find(y,x,s);
    }
}
int main() {
    int n;scanf("%d%d",&n,&B);
    len=0;memset(last,0,sizeof(last));
    for(int i=1;i<n;i++) {int x,y;scanf("%d%d",&x,&y);ins(x,y);ins(y,x);}
    if(n<B) {printf("0\n");return 0;}
    cnt=top=0;memset(sta,0,sizeof(sta));memset(belong,0,sizeof(belong));dfs(1,0);
    if(cnt==0) {cnt++;ans[1]=1;}
    find(1,0,cnt);printf("%d\n",cnt);
    for(int i=1;i<n;i++)printf("%d ",belong[i]);printf("%d\n",belong[n]);
    for(int i=1;i<cnt;i++)printf("%d ",ans[i]);printf("%d\n",ans[cnt]);
    return 0;
}