Codeforces 782C. Andryusha and Colored Balloons 搜索
Andryusha goes through a park each day. The squares and paths between them look boring to Andryusha, so he decided to decorate them.
The park consists of n squares connected with (n - 1) bidirectional paths in such a way that any square is reachable from any other using these paths. Andryusha decided to hang a colored balloon at each of the squares. The baloons' colors are described by positive integers, starting from 1. In order to make the park varicolored, Andryusha wants to choose the colors in a special way. More precisely, he wants to use such colors that if a, b and c are distinct squares that a and b have a direct path between them, and b and c have a direct path between them, then balloon colors on these three squares are distinct.
Andryusha wants to use as little different colors as possible. Help him to choose the colors!
The first line contains single integer n (3 ≤ n ≤ 2·105) — the number of squares in the park.
Each of the next (n - 1) lines contains two integers x and y (1 ≤ x, y ≤ n) — the indices of two squares directly connected by a path.
It is guaranteed that any square is reachable from any other using the paths.
In the first line print single integer k — the minimum number of colors Andryusha has to use.
In the second line print n integers, the i-th of them should be equal to the balloon color on the i-th square. Each of these numbers should be within range from 1 to k.
3
2 3
1 3
3
1 3 2
5
2 3
5 3
4 3
1 3
5
1 3 2 5 4
5
2 1
3 2
4 3
5 4
3
1 2 3 1 2
In the first sample the park consists of three squares: 1 → 3 → 2. Thus, the balloon colors have to be distinct.
In the second example there are following triples of consequently connected squares:
- 1 → 3 → 2
- 1 → 3 → 4
- 1 → 3 → 5
- 2 → 3 → 4
- 2 → 3 → 5
- 4 → 3 → 5
In the third example there are following triples:
- 1 → 2 → 3
- 2 → 3 → 4
- 3 → 4 → 5
#include<iostream> #include<cstdio> #include<cstring> #include<algorithm> #include<map> #include<vector> using namespace std; const int MAXN=2e5+100,INF=0x3f3f3f3f,MOD=1e9+7; int n,col[MAXN]; map<int,vector<int> >G; int ans; dfs(int now,int sign,int fa,int flag) { //cout<<now<<" "<<sign<<" "<<fa<<" "<<flag<<endl; int tmp=0; for(int i=0; i<G[now].size(); i++) { int to=G[now][i]; if(to==fa) continue; while(++tmp) { if(tmp!=sign&&tmp!=flag) { col[to]=tmp; if(tmp>ans) ans=tmp; dfs(to,tmp,now,sign); break; } } } } int main() { scanf("%d",&n); 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); } col[1]=1; ans=1; dfs(1,1,-1,-1); cout<<ans<<endl; for(int i=1; i<=n; i++) cout<<col[i]<<" "; cout<<endl; return 0; }