1140. 国王的遗产

#include <iostream>
#include <cstring>
#include <vector>

#define MAX 30001

using namespace std;

vector<short> edge[MAX];
bool del[MAX];
short sum[MAX];
short minNo[MAX];
short parent[MAX];
vector<int> get;

int golds, sons, leftGolds;

void init() {
	memset(del, false, sizeof(del));
	memset(sum, 1, sizeof(sum));
	int a, b;
	for(int i = 1; i <= golds - 1; i++) {
		cin >> a >> b;
		edge[a].push_back(b);
		edge[b].push_back(a);
	}
}

void dfsCount(int head, int father) {
	sum[head] = 1;
	minNo[head] = head;
	parent[head] = father;
	for(int i = 0; i < edge[head].size(); i++) {
		int tmp = edge[head][i];
		if(tmp == father || del[tmp]) {
			continue;
		}
		dfsCount(tmp, head);
		sum[head] += sum[tmp];
		minNo[head] = min(minNo[head], minNo[tmp]);
	}
}

void dfsDel(int head, int toNeglect) {
	del[head] = true;
    for(int i = 0; i < edge[head].size(); i++) {
    	int tmp = edge[head][i];
    	if(del[tmp] || tmp == toNeglect) {
    		continue;
    	}
    	dfsDel(tmp, toNeglect);
    }
    edge[head].clear();
}

void cut() {
	while(get.size() < sons - 1) {
		int mark, takenNum = 0, minimumNo = 30001;
		int root = 1;
		while(del[root]) {
			root++;
		}
		dfsCount(root, 0);
		for(int newRoot = 1; newRoot <= golds; newRoot++) {
			if(newRoot == root || del[newRoot]) {
				continue;
			}
			int part1 = sum[root] - sum[newRoot];
			int part2 = sum[newRoot];
			if(part1 == part2) {
				mark = newRoot;
				takenNum = part1;
				break;
			}
			if(part1 < part2) {
				if(part1 > takenNum || part1 == takenNum && minNo[root] < minimumNo) {
					mark = newRoot;
					takenNum = part1;
					minimumNo = minNo[root];
				}
			}
			else {
				if(part2 > takenNum || part2 == takenNum && minNo[newRoot] < minimumNo) {
					mark = newRoot;
					takenNum = part2;
					minimumNo = minNo[newRoot];
				}
			}
		}
		if(sum[mark] >= sum[root] - sum[mark]) {
			dfsDel(root, mark);
		}
		else {
			dfsDel(mark, parent[mark]);
		}
		get.push_back(takenNum);
		leftGolds -= takenNum;
	}
	get.push_back(leftGolds);
}

int main() {
	cin >> golds >> sons;
	leftGolds = golds;
	init();
	cut();
	for(int i = 0; i < sons - 1; i++) {
		cout << get[i] << ' ';
	}
	cout << get[sons-1] << endl;
	return 0;
}

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值