結果

問題 No.2301 Namorientation
コンテスト
ユーザー kys
提出日時 2023-05-13 17:06:34
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 917 ms / 3,000 ms
コード長 2,703 bytes
コンパイル時間 287 ms
コンパイル使用メモリ 82,560 KB
実行使用メモリ 159,052 KB
最終ジャッジ日時 2024-11-29 09:02:55
合計ジャッジ時間 22,294 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

def main():
    from sys import stdin, setrecursionlimit
    # setrecursionlimit(1000000)
    input = stdin.readline
    def iinput(): return int(input())
    def sinput(): return input().rstrip()
    def i0input(): return int(input()) - 1
    def linput(): return list(input().split())
    def liinput(): return list(map(int, input().split()))
    def miinput(): return map(int, input().split())
    def li0input(): return list(map(lambda x: int(x) - 1, input().split()))
    def mi0input(): return map(lambda x: int(x) - 1, input().split())
    INF = 1000000000000000000
    MOD = 1000000007

    class UnionFindTree:
        def __init__(self, initial_size:int) -> None:
            self.n_nodes = initial_size
            self.parents = [i for i in range(initial_size)]
            self.ranks = [0 for i in range(initial_size)]
            self.size = [1 for i in range(initial_size)]
            self.n_roots = initial_size
        def root(self, n:int) -> int:
            if self.parents[n] == n:
                return n
            else:
                self.parents[n] = self.root(self.parents[n])
                return self.parents[n]
        def same(self, n:int, m:int) -> bool:
            return (self.root(n) == self.root(m))
        def unite(self, n:int, m:int) -> None:
            if self.same(n, m):
                return
            self.n_roots -= 1
            n, m = self.root(n), self.root(m)
            if self.ranks[n] < self.ranks[m]:
                self.parents[n] = m
                self.size[m] += self.size[n]
            else:
                self.parents[m] = n
                self.size[n] += self.size[m]
                if self.ranks[n] == self.ranks[m]:
                    self.ranks[n] += 1
        def get_roots(self) -> set:
            return set([self.root(x) for x in range(self.n_nodes)])
        def count_roots(self) -> int:
            return self.n_roots
        def get_tree_size(self, n:int) -> int:
            return self.size[self.root(n)]
    N = iinput()
    g = [[] for _ in [0] * N]
    ans = [-1] * N
    uf = UnionFindTree(N)
    for i in range(N):
        a, b = mi0input()
        if uf.same(a, b):
            s = a
            ans[i] = '->'
            continue
        g[a].append((b, i, 0)) # <-
        g[b].append((a, i, 1)) # ->
        uf.unite(a, b)
    
    st = [s]
    seen = [False] * N
    seen[s] = True
    while st:
        u = st.pop()
        for v, idx, flg in g[u]:
            if seen[v]:
                continue
            seen[v] = True
            if flg:
                ans[idx] = '->'
            else:
                ans[idx] = '<-'
            st.append(v)

    for a in ans:
        print(a)


main()
0