ZOJ

ZOJ - 4045District Division

　　题目大意：给你n个节点的树，然后让你划分这棵数使得，每一块都恰好k个节点并且两两间是连通的，也就是划分成n/k个连通集，如果可以输出YES，并输出对应的划分，否则输出NO

　　　一开始觉得是树形dp，但不知道如何下手，看大佬的做法才恍然大悟，其实就是找子树大小刚好是k的节点，刚让它作为这一分块的根节点，它的父节点就不再记录它的子树大小，详情见代码

#include<cstdio>
#include<vector>
using namespace std; 
const int N=100118;
struct Side{
    int v,ne;
}S[2*N];
vector<int> ans;
int n,k,is,m,sn;
int head[N],size[N],book[N];
void add(int u,int v)
{
    S[sn].v=v;
    S[sn].ne=head[u];
    head[u]=sn++;
}
void fp(int u,int f)
{
    for(int i=head[u];i!=-1;i=S[i].ne)
        if(S[i].v!=f&&!book[S[i].v])
        {
            book[S[i].v]=u;
            printf(" %d",S[i].v);
            fp(S[i].v,u);
        }
}
int dfs(int u,int f)
{
    if(!is)
        return 0;
    size[u]=1;
    for(int i=head[u];i!=-1;i=S[i].ne)
    {
        if(S[i].v!=f&&!book[S[i].v])
            dfs(S[i].v,u);
        if(S[i].v!=f&&!book[S[i].v])//如果它的这个子节点没有作为划分的子树的根节点 
            size[u]+=size[S[i].v];//那么统计上它的大小 
    }
    if(size[u]>k)//如果它的子树大小大于k那么它分出一个大小k的子树后 
        return is=0;//剩下的会不足k个 
    if(size[u]==k)
    {
        ans.push_back(u);//保存作为划分的子树根节点的节点 
        book[u]=f;//并保存它的父节点，不然在输出时会PE 
    }
    return 1;
}
int main()
{    
    int t,a,b;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&k);
        ans.clear();
        for(int i=1;i<=n;i++)
            book[i]=0,head[i]=-1;
        sn=0;
        for(int i=0;i<n-1;i++)
        {
            scanf("%d%d",&a,&b);
            add(a,b),add(b,a);
        }
        is=1;
        if(dfs(1,-1))
        {
            printf("YES
");
            for(int i=0;i<ans.size();i++)
            {
                printf("%d",ans[i]);
                fp(ans[i],book[ans[i]]);
                printf("
");
            }
        }
        else
            printf("NO
");
    }
    return 0;
}