G. Privatization of Roads in Treeland--Codeforces Round #547 (Div. 3)（贪心）

题目链接：https://codeforces.com/contest/1141/problem/G

 

题意：就是给定一颗树，然后给每条边涂色（记颜色为1~r）,与某节点相关联的所有边的颜色中如果有相同的，这个节点记为unfair,要求涂色后最多有k个unfair点，求最小的r，并输出树的涂色方案；

 

 

解法：贪心

将每个节点的度数记录并排序，优先让度数大的节点成为unfair点，这样r就可以最小，所以r的答案就是deg[ n-k ]，然后dfs一遍求每个边的color

 

附代码：

#include<bits/stdc++.h>

using namespace std;
const int maxn=2e5+10;

int n,k,du[maxn],res[maxn],u,v;

struct node{
	int to,id;
	node(){}
	node(int a,int b){
		to=a;id=b;
	}
};
vector<node> vec[maxn];

void dfs(int cur,int fa,int fa_color,int num_color)
{
	int nxt=fa_color;
	for(int i=0;i<vec[cur].size();i++){
		if(vec[cur][i].to!=fa){
			nxt=(nxt%num_color)+1;
			res[vec[cur][i].id]=nxt;
			dfs(vec[cur][i].to,cur,nxt,num_color);
		}
	}
}

int main()
{
	scanf("%d %d",&n,&k);
	for(int i=1;i<n;i++){
		scanf("%d %d",&u,&v);
		vec[u].push_back(node(v,i));
		vec[v].push_back(node(u,i));
		
		du[u]++;du[v]++;
	}
	sort(du+1,du+n+1);
	int ans=du[n-k];
	printf("%d\n",ans);
	
	dfs(1,0,0,ans);
	for(int i=1;i<n;i++) printf("%d ",res[i]);
}