【天梯】python L2-023 图着色问题 (25 point(s))

图论与颜色编码问题

最新推荐文章于 2025-09-24 08:41:33 发布

原创最新推荐文章于 2025-09-24 08:41:33 发布 · 791 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#python #算法

PAT 同时被 2 个专栏收录

19 篇文章

订阅专栏

天梯

18 篇文章

订阅专栏

"该博客探讨了一种使用默认字典（defaultdict）和集合（set）在图中进行颜色编码的问题。输入包括图的顶点数（v），边数（e），以及颜色数（k）。对于每个边，博客描述了如何建立邻接关系。然后，通过检查是否存在相同颜色的相邻节点来判断是否可以进行有效染色。如果不能，输出"No"，否则输出"Yes"。该博客涉及图论和算法设计。"

在这里插入图片描述

from collections import defaultdict
v,e,k=map(int,input().split())
s=defaultdict(set)
for i in range(e):
    a,b=map(int,input().split())
    s[a].add(b)
    s[b].add(a)
n=int(input())
c=defaultdict(set)
for i in range(n):
    flag=0
    color=list(map(int,input().split()))
    if len(set(color))!=k:
        print("No")
    else:
        for j in range(v):
            for p in s[j+1]:
                if color[j]==color[p-1]:
                    flag=1
                    break
        if flag==0:
            print("Yes")
        else:
            print("No")