L2-023 图着色问题 (25 分)——测试点1 2 3分析

题目传送门

思路

测试点2：颜色必须为k种
测试点1，3过不去，大概率是dfs写错了（如果是写的dfs），不能先判断走没走过，要先判断颜色是否一样

代码

#include<bits/stdc++.h>
#define rep(i,x,y) for(int i=x; i<=y; ++i)
#define per(i,x,y) for(int i=x; i>=y; --i)
#define pushk push_back
#define popk pop_back
#define mem(a,b) memset(a,b,sizeof a)
#define PII pair<int,int>
#define ll long long
using namespace std;
const int N = 1e5+9;
vector<int> g[N];
int col[N];
bool vis[N];
int v,e,k;
bool dfs(int u) {
	vis[u]=1;
	for(auto &it : g[u]) {
		if(col[it]==col[u]) return 0;
		if(vis[it]) continue;
		if(!dfs(it)) return false;
	}
	return true;
}
int main() {


	cin>>v>>e>>k;
	rep(i,1,e) {
		int u,v;
		cin>>u>>v;
		g[u].pushk(v);
		g[v].pushk(u);
	}
	int q;
	cin>>q;
	while(q--) {
		set<int> se;
		bool ok=1;
		mem(vis,0);
		rep(i,1,v) {
			cin>>col[i];
			se.insert(col[i]);
		}
		rep(i,1,v) {
			if(vis[i]==0) {
				if(dfs(i)==false) ok=0;
			}
		}
		if(ok==false || se.size()!=k ) puts("No");
		else puts("Yes");
	}
	return 0;
}