1154 Vertex Coloring (25 分)

版本2

图的深搜

#include <cstdio>
#include <iostream>
#include <vector>
#include <cstring>
#include <queue> 
#include <unordered_set>
#include <algorithm>
using namespace std;
const int N = 1e4+5;
vector<int> G[N];
bool vis[N]; 
int n, m;
int color[N];
bool flag;
void dfs(int r){ 
	if(flag) return; //如果着色冲突则返回 
	vis[r] = true;
	for(int i = 0; i < G[r].size(); i++){
		int v = G[r][i];
		if(color[r] == color[v]) flag = true;
		if(flag) return; 
		if(!vis[v]) dfs(v);
	}
}
int main(){
	scanf("%d%d", &n, &m);
	int u, v;
	while(m--){
		scanf("%d%d", &u, &v);
		G[u].push_back(v);
		G[v].push_back(u); //无向图 
	}
	
	int k;
	scanf("%d", &k);
	while(k--){
		unordered_set<int> st;
		flag = false;
		for(int i = 0; i < n; i++) scanf("%d", color+i), st.insert(color[i]);
		memset(vis, 0, sizeof vis);
		for(int i = 0; i < n; i++){
			if(!vis[i]) dfs(i);
		}
		if(!flag) printf("%d-coloring\n", st.size());
		else puts("No");
	}
	
	return 0;
}

版本1

遍历所有边，不需要用深搜，只需要把读入的每条边再循环一遍就ok了

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e4+5;
int arr[maxn];
struct node{
	int u,v;
}E[maxn];

set<int> st;

int main(){
	int n,m,u,v;
	scanf("%d%d",&n,&m);
	
	for(int i=0; i<m; i++){
		scanf("%d%d",&u,&v);
		E[i].u = u;
		E[i].v = v;
	}
	int k;
	scanf("%d",&k);
	while(k--){
		memset(arr,-1,sizeof(arr));
		st.clear();
		for(int i=0;i<n;i++){
			scanf("%d",&arr[i]);
			st.insert(arr[i]); 
		}
		int flag=0;
		for(int i=0;i<m;i++){
			if(arr[E[i].u] == arr[E[i].v]){
				flag=1;
				break;
			}
		}
		if(flag){
			printf("No\n");
		}else 
			printf("%d-coloring\n",st.size());
	}
	
	
	return 0;
}