結果

問題 No.1023 Cyclic Tour
コンテスト
ユーザー ptotq
提出日時 2020-04-10 22:27:26
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 2,127 bytes
コンパイル時間 167 ms
コンパイル使用メモリ 81,956 KB
実行使用メモリ 388,732 KB
最終ジャッジ日時 2024-09-15 20:57:51
合計ジャッジ時間 9,830 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 12 TLE * 1 -- * 36
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline


class UnionFind(object):
    def __init__(self, n: int):
        self.nodes = [-1]*n

    def find(self, x: int) -> int:
        if self.nodes[x] < 0:
            return x
        else:
            self.nodes[x] = self.find(self.nodes[x])
            return self.nodes[x]

    def unite(self, x: int, y: int) -> bool:
        root_x, root_y, nodes = self.find(x), self.find(y), self.nodes

        if root_x != root_y:
            if nodes[root_x] > nodes[root_y]:
                root_x, root_y = root_y, root_x
            nodes[root_x] += nodes[root_y]
            nodes[root_y] = root_x

        return root_x != root_y


def kosaraju(edges, reverse_edges):
    import sys
    from operator import itemgetter
    sys.setrecursionlimit(10**7)
    v_count = len(edges)
    order = [0]*v_count
    k = [1]

    def get_order(v):
        order[v] = 1
        for dest in edges[v]:
            if order[dest] == 0:
                get_order(dest)
        order[v] = k[0]
        k[0] += 1

    def get_components(v):
        order[v] = 0
        return [v] + [_v for dest in reverse_edges[v] if order[dest] > 0 for _v in get_components(dest)]

    for v in range(v_count):
        if order[v] == 0:
            get_order(v)

    return [get_components(v) for v, _ in sorted(enumerate(order), key=itemgetter(1), reverse=True) if order[v] > 0]


if __name__ == '__main__':
    N, M = map(int, input().split())
    edges = []
    uf = UnionFind(N)

    for u, v, c in (map(int, input().split()) for _ in range(M)):
        u -= 1
        v -= 1
        if c == 1:
            if not uf.unite(u, v):
                print('Yes')
                exit()
        else:
            edges.append((u, v))

    adj = [set() for _ in range(N)]
    rev = [set() for _ in range(N)]
    for u, v in edges:
        par_u, par_v = uf.find(u), uf.find(v)
        if par_u == par_v:
            print('Yes')
            exit()
        adj[par_u].add(par_v)
        rev[par_v].add(par_u)

    res = kosaraju(adj, rev)
    if max(len(sc) for sc in res) > 1:
        print('Yes')
    else:
        print('No')
0