题意:
把有n个节点的一棵树分成n/k棵树。
保证b%k=0。
输出每棵树的节点序号。
否则输出NO。
POINT:
输出序号可以借助stack。
对于每个根节点,看每条子树还有多少节点。
因为每个子树要形成一个树(假设他们合起来为个数为k),必须要用到根节点。
所以对于每个子树的节点加起来,在加上1(根节点)。如果大于k,就NO了。
要正好等于k,才符合题意。
若小于k,那么回溯往前面合并。
#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <vector>
#include <stack>
using namespace std;
const int maxn = 1e5+33;
vector<int>G[maxn];
vector<int>a[maxn];
int ans=1;
int cnt=0;
int n,k;
stack<int> q;
int dfs(int u,int pre)
{
if(!ans) return -1;
int sum=1;
q.push(u);
for(int i=0;i<G[u].size();i++){
int v=G[u][i];
if(v==pre) continue;
int to=dfs(v,u);
sum+=to;
}
if(sum>k){
ans=0;
return -1;
}
if(k==sum){
cnt++;
while(sum--){
a[cnt].push_back(q.top());
q.pop();
}
sum=0;
}
return sum;
}
int main()
{
int T;
scanf("%d",&T);
while(T--){
scanf("%d%d",&n,&k);
ans=1;
cnt=0;
for(int i=1;i<=n;i++){
G[i].clear();
a[i].clear();
}
for(int i=1;i<n;i++){
int u,v;scanf("%d%d",&u,&v);
G[u].push_back(v);
G[v].push_back(u);
}
dfs(1,-1);
if(!ans){
printf("NO\n");
}else{
printf("YES\n");
for(int i=1;i<=cnt;i++){
for(int j=0;j<k;j++){
printf("%d%c",a[i][j],j==k-1?'\n':' ');
}
}
}
}
}