ZOJ 4045(并查集+dfs)

题目链接：点击打开链接

题目思路：比赛的时候是找根节点，然后退后，结果一直tle...................................难受.jpg

看了学长代码。学长代码是直接深搜。从1开始，不停搜，搜到根节点以后开始退一步，然后把根节点爸爸的儿子都带进来(其实我当时的思路也是这样的QAQ)，sum来记录这个节点有几个儿子(当时想了半天怎么分辨谁是儿子谁是爸爸，后来发现只要从根节点往回退，就肯定找到都是爸爸..跟这个好像有异曲同工之妙..但还是没搞出来..)如果到了k，那就可以另起炉灶找新的分区了，如果超过了k，那就凉凉，输出no。最后再来个结构体，一个放爸爸(并查集里的)，一个放编号，然后根据爸爸排序，就能把相同分区的放一起了，然后输出就好，学长真优秀

以下是代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
#define inf 0x3f3f3f3f
#define MAXN 100005
#define rep(i,a,b) for(int i=a;i<=b;++i)
int cnt,head[MAXN],pre[MAXN],vis[MAXN],k,flag,n;
struct node{
    int next,to;
}edge[MAXN*2];
struct ss{
    int id,fa;
}a[MAXN];
void add(int x,int y){
    edge[cnt].to=y;
    edge[cnt].next=head[x];
    head[x]=cnt++;
}
int Find(int x){
    return pre[x]==x?x:pre[x]=Find(pre[x]);
}
int cmp(ss a,ss b){
    return a.fa<b.fa;
}
int dfs(int x){
    vis[x]=1;
    int sum=1;
    for(int i=head[x];i!=-1;i=edge[i].next){
        int z=edge[i].to;
        if(vis[z])continue;
        int num=dfs(z);
        if(num<k){
            sum+=num;
            int xx=Find(x),yy=Find(z);
            pre[xx]=yy;
        }
    }
    if(sum>k)flag=0;
    return sum;
}
int main(){
    int t,x,y,n;
    scanf("%d",&t);
   while(t--){
        memset(vis,0,sizeof(vis));
        memset(head,-1,sizeof(head));
        cnt=0,flag=1;
        scanf("%d%d",&n,&k);
        rep(i,1,n)pre[i]=i;
        rep(i,1,n-1){
            scanf("%d%d",&x,&y);
            add(x,y);
            add(y,x);
        }
        int z=dfs(1);
        if(flag){
            printf("YES\n");
            rep(i,1,n){
                a[i].id=i;
                a[i].fa=Find(i);
            }
            sort(a+1,a+n+1,cmp);
            a[n+1].fa=-1;
            rep(i,1,n){
                printf("%d",a[i].id);
                if(a[i].fa!=a[i+1].fa)printf("\n");
                else printf(" ");
            }
        }
        else{
            printf("NO\n");
        }
    }
    return 0;
}