ZOJ Monthly, June 2018 - I District Division

点击打开链接

题意：

把有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':' ');
				}
			}
		}

	}

}