acwing搜索与图论（三）染色法

一缕叶2025-01-28 10:56

复制代码

#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;


const  int N=100010,M=200020;

int n,m;
int h[N],ne[M],e[M],idx;
int color[N];

void add(int a,int b){
	e[idx]=b,ne[idx]=h[a],h[a]=idx++;
	
}


bool dfs(int u,int c)
{
	color[u]=c;
	for(int i=h[u];i!=-1;i=ne[i]){
		int j=e[i];
		if(!color[j]){
			if(!dfs(j,3-c))return false;
			
			
			
		}
		else if(color[j]==c)return false;
		
	}
	
	return true;
}
int main(){
	cin>>n>>m;
	memset(h,-1,sizeof h);
	while(m--){
		int a,b;
		cin>>a>>b;
		add(a,b),add(b,a);
	}
	bool flag=true;
	for(int i=1;i<=n;i++){
		if(!color[i]){
			if(!dfs(i,1)){
				
				 flag=false;
				break;
			}
		}
	}
	
	
	
	
	if(flag){
		puts("YES");
	}
	else puts("NO");
	
	
}

测试数据：

复制代码

其实这个代码挺简单的，存储图的话就两种方式邻接矩阵或者邻接表，邻接表方便dfs深度优先，深度，用深度优先进行判断相邻的两个结点可否染异色，不可以则false，然后将所有点遍历一边。