题目描述
给出 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");
}
}
}