L2-023 图着色问题 (25 分)

L2-023 图着色问题 (25 分)
注意：顶点的编号是从1开始的 ，同时要判断给定的给出的颜色数是否等于k

#include<bits/stdc++.h>
#include<iostream>
using namespace std;
int v,e,k,x,y,n;
vector<int> G[505];
int arr[505];
set<int> s;
bool judge(){
	for(int i=1;i<=v;i++){  //顶点的编号是从1开始的 
		for(int j=0;j<G[i].size();j++){
			int temp = G[i][j];
			if(arr[i]==arr[temp]) return false;
		}
	}
	return true;
}
int main(){
	
	cin>>v>>e>>k;
	
	for(int i=0;i<e;i++){
		cin>>x>>y;
		G[x].push_back(y);
		G[y].push_back(x);
	}
	cin>>n;
	
	for(int i=0;i<n;i++){
		memset(arr,0,sizeof(arr));
		s.clear();
		for(int i=1;i<=v;i++){
			cin>>arr[i];
			s.insert(arr[i]);
		}
		if(s.size()==k && judge()) cout<<"Yes"<<endl;
		else cout<<"No"<<endl;
	}
	
	return 0;
}