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

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")