Codeforces Round #403 C. Andryusha and Colored Balloons

題目鏈接：點擊打開鏈接

題意：給出n個點，（n-1）條無向邊，問能否將點染色使所有點與其鄰接的點以及距離爲1的點的顏色不同（如：a與b存在一邊，b與c存在一邊，則a，b，c三點顏色應不同）

題解：將圖看做一棵樹，對於一個節點的子節點我們先將選一個節點進行染色，染色的顏色選擇則是從1開始不斷++一直到與其父節點以及父節點的父節點顏色不同爲止，然後再對這個子節點進行進一步深搜，因爲這個子節點的子節點不受這個子節點的兄弟的顏色影響所以可以繼續深搜，搜完第一個子節點後則對第二個子節點進行染色，第二個子節點的顏色則是先++然後一直++到和其父節點以及父節點的父節點不同爲止，然後再做與第一個子節點相同的深搜，以此類推即可。

#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<cmath>
#include<stdlib.h>
#include <string.h>
#include<queue>
#include<set>
#include<map>
#include<stack>
#include<time.h>
using namespace std;
#define MAX_N 200005
#define inf 0x3f3f3f3f
#define LL long long
#define ull unsigned long long
const LL INF = 1e18;
const int mod = 1e8+7;
typedef pair<double, int>P;

int n;
int mx;
vector<int>G[MAX_N];
int color[MAX_N];
void dfs(int u, int fa)
{
    int c = 0;
    for(int i=0; i<G[u].size(); i++) {
        if(G[u][i] != fa) {
            c++;
            while(c == color[u] || c == color[fa])
                c++;
            color[G[u][i]] = c;
            mx = max(mx, c);
        }
    }
    for(int i=0; i<G[u].size(); i++) {
        if(G[u][i] != fa) {
            dfs(G[u][i], u);
        }
    }
}
int main()
{
    memset(color, 0, sizeof(color));
    mx = 0;
    cin >> n;
    for(int i=1; i<n; i++) {
        int a, b;
        scanf("%d%d", &a, &b);
        G[a].push_back(b);
        G[b].push_back(a);
    }
    color[1] = 1;
    dfs(1, 1);
    printf("%d\n", mx);
    for(int i=1; i<=n; i++)
        printf("%d ",color[i]);
}