結果

問題 No.2630 Colorful Vertices and Cheapest Paths
ユーザー ゼットゼット
提出日時 2024-02-16 22:08:50
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,554 ms / 2,500 ms
コード長 1,204 bytes
コンパイル時間 392 ms
コンパイル使用メモリ 82,592 KB
実行使用メモリ 123,988 KB
最終ジャッジ日時 2024-09-28 20:26:40
合計ジャッジ時間 21,318 ms
ジャッジサーバーID
(参考情報)
judge2 / judge4
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 560 ms
81,900 KB
testcase_01 AC 1,163 ms
97,056 KB
testcase_02 AC 1,084 ms
109,056 KB
testcase_03 AC 63 ms
74,892 KB
testcase_04 AC 40 ms
62,492 KB
testcase_05 AC 67 ms
76,700 KB
testcase_06 AC 630 ms
81,160 KB
testcase_07 AC 943 ms
117,944 KB
testcase_08 AC 850 ms
105,900 KB
testcase_09 AC 924 ms
118,640 KB
testcase_10 AC 1,168 ms
118,808 KB
testcase_11 AC 1,159 ms
119,956 KB
testcase_12 AC 1,150 ms
122,012 KB
testcase_13 AC 1,217 ms
123,988 KB
testcase_14 AC 1,151 ms
115,672 KB
testcase_15 AC 1,499 ms
80,032 KB
testcase_16 AC 1,540 ms
79,764 KB
testcase_17 AC 1,554 ms
79,892 KB
testcase_18 AC 669 ms
89,364 KB
testcase_19 AC 823 ms
100,440 KB
testcase_20 AC 718 ms
95,232 KB
testcase_21 AC 921 ms
117,584 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

class unif:
  def __init__(self,n):
    self.pare=[-1]*n
    self.size=[1]*n
  def root(self,x):
    while self.pare[x]!=-1:
      x=self.pare[x]
    return x
  def unite(self,u,v):
    rootu=self.root(u)
    rootv=self.root(v)
    if rootu!=rootv:
      if self.size[rootu]>=self.size[rootv]:
        self.pare[rootv]=rootu
        self.size[rootu]+=self.size[rootv]
      else:
        self.pare[rootu]=rootv
        self.size[rootv]+=self.size[rootu]
  def same(self,s,t):
    return self.root(s)==self.root(t)
N,M=map(int,input().split())
L=[]
for i in range(M):
  a,b=map(int,input().split())
  L.append((a-1,b-1))
C=list(map(int,input().split()))
for i in range(N):
  C[i]-=1
W=list(map(int,input().split()))
Q=int(input())
result=[10**15]*Q
q=[]
for _ in range(Q):
  u,v=map(int,input().split())
  q.append((u-1,v-1))
for bit in range(1,2**10):
  cost=0
  for k in range(10):
    if (bit>>k)&1:
      cost+=W[k]
  Z=unif(N)
  for i in range(M):
    a,b=L[i][:]
    if (bit>>C[a])&1 and (bit>>C[b])&1:
      Z.unite(a,b)
  for i in range(Q):
    u,v=q[i][:]
    if Z.same(u,v):
      result[i]=min(result[i],cost)
for i in range(Q):
  if result[i]<10**14:
    print(result[i])
  else:
    print(-1)
0