Codeforces Round #547 (Div. 3) 贪心

https://codeforces.com/contest/1141/problem/G

题意

在一棵有n个点的树上给边染色,连在同一个点上的边颜色不能相同,除非舍弃掉这个点,问最少需要多少种颜色来染一棵树

题解

• 选择弃掉度数最高的k个点,然后第k+1个点的度数就是答案

代码

#include<bits/stdc++.h>
#define N 200005
#define pb push_back
using namespace std;
int n,k,u,v,in[N],c[N],m,i;
struct node{
    int e,v;
};
vector<node>g[N];

bool cmp(int x,int y){
    return x>y;
}
void dfs(int u,int fa,int cl){
    for(int i=0;i<g[u].size();i++){
        int v=g[u][i].v,id=g[u][i].e;if(v==fa)continue;
        if(cl>m)cl-=m;
        c[id]=cl++;
        dfs(v,u,cl);
    }
}

int main(){
    cin>>n>>k;
    for(i=1;i<=n-1;i++){
        scanf("%d%d",&u,&v);
        in[u]++;in[v]++;
        g[u].pb({i,v});
        g[v].pb({i,u});
    }
    sort(in+1,in+n+1,cmp);
    m=in[k+1];
    dfs(1,0,1);
    printf("%d\n",m);
    for(i=1;i<=n-1;i++){
        printf("%d ",c[i]);
    }
}