結果
問題 | No.1293 2種類の道路 |
ユーザー | 👑 SPD_9X2 |
提出日時 | 2020-11-20 22:18:42 |
言語 | PyPy3 (7.3.13) |
結果 |
AC
|
実行時間 | 337 ms / 2,000 ms |
コード長 | 2,021 bytes |
コンパイル時間 | 263 ms |
コンパイル使用メモリ | 87,272 KB |
実行使用メモリ | 111,768 KB |
最終ジャッジ日時 | 2023-09-30 19:28:03 |
合計ジャッジ時間 | 6,346 ms |
ジャッジサーバーID (参考情報) |
judge15 / judge14 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 76 ms
70,692 KB |
testcase_01 | AC | 71 ms
71,080 KB |
testcase_02 | AC | 71 ms
70,968 KB |
testcase_03 | AC | 72 ms
70,804 KB |
testcase_04 | AC | 72 ms
70,924 KB |
testcase_05 | AC | 72 ms
70,688 KB |
testcase_06 | AC | 70 ms
70,804 KB |
testcase_07 | AC | 71 ms
70,828 KB |
testcase_08 | AC | 71 ms
70,972 KB |
testcase_09 | AC | 334 ms
98,704 KB |
testcase_10 | AC | 332 ms
98,672 KB |
testcase_11 | AC | 336 ms
98,776 KB |
testcase_12 | AC | 337 ms
99,164 KB |
testcase_13 | AC | 326 ms
98,552 KB |
testcase_14 | AC | 228 ms
99,584 KB |
testcase_15 | AC | 232 ms
99,600 KB |
testcase_16 | AC | 268 ms
104,028 KB |
testcase_17 | AC | 266 ms
104,604 KB |
testcase_18 | AC | 225 ms
103,948 KB |
testcase_19 | AC | 237 ms
111,768 KB |
testcase_20 | AC | 236 ms
111,656 KB |
testcase_21 | AC | 212 ms
86,848 KB |
testcase_22 | AC | 213 ms
87,176 KB |
testcase_23 | AC | 204 ms
87,016 KB |
ソースコード
""" 道路の種類ごとにunion-findする 後は自動車専用道路と歩行者専用道路のリンクを考える… 結局普通に数えたのではN**2になってしまう まとめて数える方法が必須 歩行者側だけ繋いでおく…? 後は連結成分だけ考えればいいはず 自動車専用をつなぐと、そこの間の組が通行可能になる →まず歩行者側をつないで連結成分の親だけのグラフにする ん~~~ 超頂点を作る X1,X2 フロー!? DはX1側を交互につなぐ 二部グラフにはなるけど… 下側の連結成分をまとめておく 上側もまとめておく 上側を連結成分ごとに計算する 下にある連結成分は高々N個なので dictで重複を管理しつつやればO(NlogN)で解ける """ from sys import stdin def uf_find(n,p): ufl = [] while p[n] != n: ufl.append(n) n = p[n] for i in ufl: p[i] = n return n def uf_union(a,b,p,rank): ap = uf_find(a,p) bp = uf_find(b,p) if ap == bp: return False else: if rank[ap] > rank[bp]: p[bp] = ap rank[ap] += rank[bp] else: p[ap] = bp rank[bp] += rank[ap] return True N,D,W = map(int,stdin.readline().split()) ab = [] for i in range(D): a,b = map(int,stdin.readline().split()) ab.append((a-1,b-1)) cd = [] for i in range(W): c,d = map(int,stdin.readline().split()) cd.append((c-1,d-1)) p1 = [i for i in range(N)] rank1 = [1] * N for a,b in ab: uf_union(a,b,p1,rank1) p2 = [i for i in range(N)] rank2 = [1] * N for c,d in cd: uf_union(c,d,p2,rank2) ans = 0 ch = [ {} for i in range(N) ] for i in range(N): underP = uf_find(i,p2) ch[uf_find(i,p1)][underP] = rank2[underP] #print (ch) ans = 0 for i in range(N): if len(ch[i]) > 0: tsum = 0 for key in ch[i]: tsum += ch[i][key] ans += rank1[i] * tsum print (ans - N)