题意

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

思路

因为是连通的问题,所以用到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;
}
学如逆水行舟,不进则退
Logo

权威|前沿|技术|干货|国内首个API全生命周期开发者社区

更多推荐