結果
問題 | No.488 四角関係 |
ユーザー |
![]() |
提出日時 | 2020-05-21 11:17:43 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 686 ms / 5,000 ms |
コード長 | 1,657 bytes |
コンパイル時間 | 114 ms |
コンパイル使用メモリ | 12,928 KB |
実行使用メモリ | 11,136 KB |
最終ジャッジ日時 | 2024-10-01 23:55:09 |
合計ジャッジ時間 | 4,983 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 22 |
ソースコード
import sysfrom itertools import combinationsdef input(): return sys.stdin.readline().strip()def list2d(a, b, c): return [[c] * b for i in range(a)]def list3d(a, b, c, d): return [[[d] * c for j in range(b)] for i in range(a)]def list4d(a, b, c, d, e): return [[[[e] * d for j in range(c)] for j in range(b)] for i in range(a)]def ceil(x, y=1): return int(-(-x // y))def INT(): return int(input())def MAP(): return map(int, input().split())def LIST(N=None): return list(MAP()) if N is None else [INT() for i in range(N)]def Yes(): print('Yes')def No(): print('No')def YES(): print('YES')def NO(): print('NO')sys.setrecursionlimit(10 ** 9)INF = 10 ** 19MOD = 10 ** 9 + 7EPS = 10 ** -10def warshall_floyd(graph: list) -> list:""" ワーシャルフロイド(頂点数, 隣接行列(0-indexed)) """from copy import deepcopyN = len(graph)res = deepcopy(graph)for i in range(N):res[i][i] = 0for k in range(N):for i in range(N):for j in range(N):res[i][j] = min(res[i][j], res[i][k] + res[k][j])for i in range(N):if res[i][i] < 0:return []return resN, M = MAP()G = list2d(N, N, INF)for i in range(M):a, b = MAP()G[a][b] = 1G[b][a] = 1wf = warshall_floyd(G)ans = 0for comb in combinations(range(N), 4):for i in range(4):C = [0] * 3for j in range(4):if i == j:continuea, b = comb[i], comb[j]if wf[a][b] <= 2:C[wf[a][b]] += 1if C[1] != 2 or C[2] != 1:breakelse:ans += 1print(ans)