結果

問題 No.2630 Colorful Vertices and Cheapest Paths
ユーザー ニックネーム
提出日時 2024-02-16 22:19:15
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 993 ms / 2,500 ms
コード長 960 bytes
コンパイル時間 294 ms
コンパイル使用メモリ 83,100 KB
実行使用メモリ 82,892 KB
最終ジャッジ日時 2024-09-28 20:54:40
合計ジャッジ時間 15,474 ms
ジャッジサーバーID
(参考情報)
judge2 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

diff #

class UF:
    def __init__(self,n):
        self.p = [-1]*n
    def f(self,x):
        if self.p[x]<0: return x
        else: self.p[x] = self.f(self.p[x]); return self.p[x]
    def u(self,x,y):
        x = self.f(x); y = self.f(y)
        if x==y: return
        if -self.p[x]<-self.p[y]: x,y = y,x
        self.p[x] += self.p[y]; self.p[y] = x
    def s(self,x,y): return self.f(x) == self.f(y)
n,m = map(int,input().split())
ab = [[v-1 for v in map(int,input().split())] for _ in range(m)]
c = [v-1 for v in map(int,input().split())]
w = list(map(int,input().split()))
q = int(input()); k = 10; d = [sum(w)+1]*q
uv = [[v-1 for v in map(int,input().split())] for _ in range(q)]
for s in range(1<<k):
    x = 0; uf = UF(n)
    for i,v in enumerate(w):
        if s>>i&1: x += v
    for a,b in ab:
        if s>>c[a]&1 and s>>c[b]&1: uf.u(a,b)
    for i,(u,v) in enumerate(uv):
        if uf.s(u,v): d[i] = min(d[i],x)
for v in d: print(v if v<=sum(w) else -1)
0