結果
| 問題 |
No.1473 おでぶなおばけさん
|
| コンテスト | |
| ユーザー |
学ぶマン
|
| 提出日時 | 2025-02-25 19:53:24 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
AC
|
| 実行時間 | 863 ms / 2,000 ms |
| コード長 | 4,067 bytes |
| コンパイル時間 | 596 ms |
| コンパイル使用メモリ | 82,192 KB |
| 実行使用メモリ | 134,244 KB |
| 最終ジャッジ日時 | 2025-02-25 19:53:55 |
| 合計ジャッジ時間 | 27,855 ms |
|
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 2 |
| other | AC * 47 |
ソースコード
import sys, math
sys.setrecursionlimit(10**8)
sys.set_int_max_str_digits(0)
INF = 10**18
MOD = 998244353
from bisect import bisect_left, bisect_right
from collections import deque, defaultdict, Counter
from itertools import product, combinations, permutations, groupby, accumulate
from heapq import heapify, heappop, heappush
def I(): return sys.stdin.readline().rstrip()
def II(): return int(sys.stdin.readline().rstrip())
def IS(): return sys.stdin.readline().rstrip().split()
def MII(): return map(int, sys.stdin.readline().rstrip().split())
def LI(): return list(sys.stdin.readline().rstrip())
def TII(): return tuple(map(int, sys.stdin.readline().rstrip().split()))
def LII(): return list(map(int, sys.stdin.readline().rstrip().split()))
def LSI(): return list(map(str, sys.stdin.readline().rstrip().split()))
def GMI(): return list(map(lambda x: int(x) - 1, sys.stdin.readline().rstrip().split()))
def kiriage(a, b): return (a+b-1)//b
class UnionFind():
def __init__(self, n):
self.n = n
self.parents = [-1] * n
self.link = [i for i in range(n)]
#self.L = list(range(n))
#self.R = list(range(n))
self.score = [0] * n
def find(self, x):
if self.parents[x] < 0:
return x
else:
self.parents[x] = self.find(self.parents[x])
return self.parents[x]
def union(self, x, y):
rootx = self.find(x)
rooty = self.find(y)
if rootx == rooty: # すでに same ならなにもしない!
return
if self.parents[rootx] > self.parents[rooty]: # -3 > -4 のようなケース(前者の方が軍勢が少ない)
rootx, rooty = rooty, rootx # rootx = 多い, rooty = 少ない の順番に変更
# 統合(rootx が新しい棟梁, rooty は軍門に下った)
self.parents[rootx] += self.parents[rooty] # 多い方(rootx)に併合する
self.parents[rooty] = rootx
self.link[x], self.link[y] = self.link[y], self.link[x]
#self.L[rootx] = min(self.L[rootx], self.L[rooty])
#self.R[rootx] = max(self.R[rootx], self.R[rooty])
self.score[rootx] += self.score[rooty]
self.score[rooty] = 0
def size(self, x): # O(1)
return -self.parents[self.find(x)]
def same(self, x, y): # O(1)
return self.find(x) == self.find(y)
def members(self, x): # O(K: 集合の要素数) 最小で1、最大でN
res = [x]
now = self.link[x]
while now != x:
res.append(now)
now = self.link[now]
return res
def roots(self): # O(N)
return [i for i, x in enumerate(self.parents) if x < 0]
def group_count(self): # roots() の呼び出しは全頂点の par をチェックする=遅いため、N回使うと TLE する
return len(self.roots())
def all_group_members(self):
group_members = defaultdict(list)
for member in range(self.n):
group_members[self.find(member)].append(member)
return group_members
def __str__(self):
return '\n'.join(f'{r}: {m}' for r, m in self.all_group_members().items())
N, M = map(int, input().split())
G = [list() for _ in range(N)]
edges = []
for i in range(M):
u, v, w = map(int, input().split())
u -= 1
v -= 1
edges.append((w, u, v))
# コスト単位で大きい辺から追加していく
# 最初に uf.same(0, N-1) となったときの値が ans1
uf = UnionFind(N)
edges.sort(reverse=True)
ans1 = -1
flag = False
for w, u, v in edges:
if flag and w != ans1: # ans1 と同じコストの辺は回収する
break
# 辺を追加 -> check の順
uf.union(u, v)
G[u].append(v)
G[v].append(u)
if not flag and uf.same(0, N - 1):
ans1 = w
flag = True
# BFS
dist = [-1] * N
dist[0] = 0
que = deque([0])
while que:
pos = que.popleft()
for nex in G[pos]:
if dist[nex] == -1:
dist[nex] = dist[pos] + 1
que.append(nex)
ans2 = dist[-1]
print(ans1, ans2)
学ぶマン