Технокубок 2017 problem: (C) Andryusha and Colored Balloons【bfs】
题意给你几个连通的节点,现在想让你染色,且每个节点的颜色与和这个节点的父节点和爷爷节点的颜色不同,想让你计算最少需要用到几个颜色思路因为是连通的问题,所以用到vector容器,用push_back()压入,然后因为对节点的颜色有要求,所以bfs,从节点1开始,染色每次都从颜色1开始搜索并且判断其父节点和爷爷节点的颜色是否相同,若相同则颜色+1继续判断,直到不相同为止,由此可以得到最少需要用到...
·
题意
给你几个连通的节点,现在想让你染色,且每个节点的颜色与和这个节点的父节点和爷爷节点的颜色不同,想让你计算最少需要用到几个颜色
思路
因为是连通的问题,所以用到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;
}
学如逆水行舟,不进则退
更多推荐
已为社区贡献2条内容
所有评论(0)