求是否给出的颜⾊数量
==k
,并且任意两个点之间颜⾊都不同。
用vector来存 与该顶点在一条边上的所有顶点,然后双重遍历即可。
#include<bits/stdc++.h>
using namespace std;
const int maxn=510;
// int g[maxn][maxn];
vector<int> tu[maxn];
int main(){
int v,e,k;
cin>>v>>e>>k;
for(int i=0;i<e;i++){
int x,y;
cin>>x>>y;
// g[x][y]=g[y][x]=1; 并没有什么卵用
tu[x].push_back(y);tu[y].push_back(x);
}
int m;cin>>m;
for(int i=0;i<m;i++){
vector<int> color(v+1);
set<int> s;
for(int j=1;j<=v;j++){
cin>>color[j];s.insert(color[j]);
}
if(s.size()!=k) printf("No\n");
else{
bool flag=true;
for(int j=1;j<=v;j++){
for(int x=0;x<tu[j].size();x++){
if(color[j]==color[tu[j][x]]){
flag=false;break;
}
}
}
if(flag) printf("Yes\n");
else printf("No\n");
}
}
return 0;
}