HNUCM-1625：Tree

题目描述
给出 N 个点的树和 K，问能否把树划分成 N/K个连通块，且每个连通块的点数都是 K。
输入
第 1 行，1 个整数 T，表示数据组数。接下来 T 组数据，对于每组数据：
第 1 行，2 个整数 N，K。 接下来 (N −1) 行，每行 2 个整数Ai,Bi,表示边
(Ai,Bi)点用 1,2,…,N 编号。
（1≤T≤10,1≤N,K≤10^5）
输出
对于每组数据，输出 YES或 NO。
样例输入
2
4 2
1 2
2 3
3 4
4 2
1 2
1 3
1 4
样例输出
YES
NO

分析
这道题属于连通块问题，问题的正解就是要能够遍历一个点，以这个点为根的情况下，结点的个数，如果结点的个数等于k，则可以成为一个连通块，如果结点的个数大于k，那么肯定输出NO，当结点个数小于k时，并到其他根上，直到满足条件为止。

代码如下

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+5;
struct node{
    int to,next;
}e[maxn*2];
int head[maxn],cnt;
int n,k;
bool flag;
inline void init(){
    memset(head,-1,sizeof(head));
    cnt=0;
}
inline void add(int x,int y){
    e[cnt].to=y;
    e[cnt].next=head[x];
    head[x]=cnt++;
}
inline int dfs(int id,int root){
    if(!flag)return 0;
    int num=1;
    for(int i=head[id];i!=-1;i=e[i].next){
        int to=e[i].to;
        if(to!=root)num+=dfs(to,id);
    }
    if(num>k){flag=false;return 0;}
    //printf("%d %d\n",id,num);
    return num%k;
}
int main(){
    int t;cin>>t;
    while(t--){
        init();
        scanf("%d%d",&n,&k);
        for(int i=1;i<n;++i){
            int x,y;scanf("%d%d",&x,&y);
            add(x,y),add(y,x);
        }
        if(n%k!=0)printf("NO\n");
        else{
            flag=true;
            dfs(1,0);
            if(!flag)printf("NO\n");
            else printf("YES\n");
        }
    }
}