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;
}