Технокубок 2017 problem: (C) Andryusha and Colored Balloons【bfs】

题意

给你几个连通的节点，现在想让你染色，且每个节点的颜色与和这个节点的父节点和爷爷节点的颜色不同，想让你计算最少需要用到几个颜色

思路

因为是连通的问题，所以用到vector容器，用push_back()压入，然后因为对节点的颜色有要求，所以bfs，从节点1开始，染色每次都从颜色1开始搜索并且判断其父节点和爷爷节点的颜色是否相同，若相同则颜色+1继续判断，直到不相同为止，由此可以得到最少需要用到几个颜色，并且记录每个颜色对应的节点。

code

#include<bits/stdc++.h>
#define endl '\n'
using namespace std;
const int maxn=2e5+5;
int n,cnt;
vector<int> G[maxn];
int fa[maxn];
int ans[maxn];
void bfs(){
    cnt=1;
    queue<int> q;
    fa[1]=1;
    ans[1]=1;
    q.push(1);
    while(!q.empty()){
        int u=q.front();
        q.pop();
        int co=1;
        for(auto v:G[u]){
            if(!ans[v]){
                while(ans[u]==co||ans[fa[u]]==co) co++;
                ans[v]=co;
                cnt=max(cnt,co);
                fa[v]=u;
                co++;
                q.push(v);
            }
        }
    }
}
int main(){
    cin>>n;
    for(int i=0;i<n-1;i++){
        int u,v;
        cin>>u>>v;
        G[u].push_back(v);
        G[v].push_back(u);
    }
    bfs();
    cout<<cnt<<endl;
    for(int i=1;i<n;i++)
        cout<<ans[i]<<" ";
        cout<<ans[n]<<endl;
    return 0;
}

学如逆水行舟，不进则退