結果

問題 No.2301 Namorientation
コンテスト
ユーザー i_taku
提出日時 2023-05-18 10:25:58
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,727 bytes
コンパイル時間 273 ms
コンパイル使用メモリ 82,560 KB
実行使用メモリ 345,264 KB
最終ジャッジ日時 2024-12-16 02:39:13
合計ジャッジ時間 21,231 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 8 WA * 22
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline
INF = float('inf')
sys.setrecursionlimit(10**7)


def main():
    N = int(input())
    AB = []
    g = [[] for _ in range(N)]
    for _ in range(N):
        a, b = map(int, input().split())
        a, b = a - 1, b - 1
        AB.append((a, b))
        g[a].append(b)
        g[b].append(a)

    def dfs(u, prev, target=-1):
        for v in g[u]:
            if v in (prev, target):
                continue
            if distance[v] == INF:
                distance[v] = distance[u] + 1
                dfs(v, u)
            elif distance[v] < distance[u]:
                is_loop[prev] = is_loop[u] = is_loop[v] = True
                if target == -1:
                    target = v
                dfs(prev, u, target)

    distance = [INF] * N
    distance[0] = 0
    is_loop = [False] * N
    dfs(0, -1)

    def dfs2(u, prev=-1):
        for v in g[u]:
            if v == prev:
                continue
            if is_loop[v] and distance[v] == INF:
                distance[v] = distance[u] + 1
                dfs2(v, u)
            elif distance[v] == INF:
                distance[v] = distance[u] - 1
                dfs2(v, u)

    distance = [INF] * N
    # ループからdfs
    for i in range(N):
        if is_loop[i] and distance[i] == INF:
            distance[i] = 1 << 18
            dfs2(i)

    # 答え出力
    for a, b in AB:
        if is_loop[a] and is_loop[b]:
            if distance[b] - distance[a] == 1 or distance[a] - distance[b] > 1:
                print('->')
            else:
                print('<-')
        else:
            if distance[a] < distance[b]:
                print('->')
            else:
                print('<-')


main()
0