给定n个顶点的图,要给图上每个顶点染色且相邻顶点颜色不同。问是否最多用2种颜色进行染色,保证没有重边或自环
1<=n<=1000;
使用dfs进行遍历。因为只有两种颜色,可以设为1和-1.每下一层加一个负号。
我们使用邻接表来存。
const int MAX=1005;
int V,E;//V为点数,E为边数
//表示有MAX个vector,存有与该点链接的点,下标表示第几个点
vector<int>G[MAX];
//用来判断是否染过色
color[MAX];
int main(){
int s,t;//s为起始点,t为终止点
cin>>V>>E;
for(int i=0;i<V;i++)
color[i]=0;
for(int i=0;i<E;i++){
cin>>s>>t;
G[s].push_back(t);
}
solve(); //操作
return 0;
}
使用solve函数调用dfs判断是否染过色,因为有可能是森林,所以需要每个点都遍历。如果是连通图,则用一次dfs则可遍历所有点。
void solve(){
for(int i=0;i<V;i++){
if(color[i]==0)
//一次遍历后还没染色的点是根节点
if(!dfs(i,1)){
cout<<"NO"<<endl;
return;
}
}
cout<<"YES"<<endl;
bool dfs(int v,int c){
color[v]=c;
for(int i=0;i<G[v].size();i++){
//染过色
if(color[G[v][i]]==c) return false;
//如果没染过色,再判读下一个点,有一个点跟同图的其他节点同色返回false;
if(color[G[v][i]==0&&!dfs(G[v][i],-c)) return false;
}
//同一个连通图中所有点都染过色了
return true;