题目链接:PTA | 程序设计类实验辅助教学平台
输入样例:
6 8 3
2 1
1 3
4 6
2 5
2 4
5 4
5 6
3 6
4
1 2 3 3 1 2
4 5 6 6 4 5
1 2 3 4 5 6
2 3 4 2 3 4
输出样例:
Yes
Yes
No
No
分析: 我写这道题目的原因就是这道题目有一个细节在考试的时候没有发现,就是题目中说是否用k种颜色为V中的每一个顶点分配一种颜色,使得不会有两个相邻顶点具有同一种颜色,我理解成了只要是小于等于k种颜色染色成功均算入合法,没想到题目中的意思是恰好使用k种颜色进行染色,就这一个地方,一直没找出来,所以这道题目一直没能拿满分
这道题目只需要一开始把边存下,然后依次判断每条边两端的端点颜色是否相同即可,最后还需要特判一下颜色总数不等于k的情况。
下面是代码:
#include<cstdio>
#include<iostream>
#include<cstring>
#include<vector>
#include<algorithm>
#include<map>
#include<cmath>
#include<queue>
#include<set>
using namespace std;
const int N=1e6+10;
int h[N],e[N],ne[N],idx,c[N];
void add(int x,int y)
{
e[idx]=y;
ne[idx]=h[x];
h[x]=idx++;
}
int main()
{
int n,m,k;
cin>>n>>m>>k;
memset(h,-1,sizeof h);
for(int i=1;i<=m;i++)
{
int u,v;
scanf("%d%d",&u,&v);
add(u,v);
}
int q;
cin>>q;
while(q--)
{
set<int>s;
bool flag=true;
for(int i=1;i<=n;i++)
scanf("%d",&c[i]),s.insert(c[i]);
for(int i=1;i<=n;i++)
{
for(int j=h[i];j!=-1;j=ne[j])
if(c[e[j]]==c[i]) flag=false;
}
if(flag&&s.size()==k) printf("Yes");
else printf("No");
if(q) puts("");
}
return 0;
}