結果

問題 No.101 ぐるぐる!あみだくじ!
ユーザー rpy3cpp
提出日時 2015-07-18 00:55:24
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
RE  
(最新)
AC  
(最初)
実行時間 -
コード長 793 bytes
コンパイル時間 265 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 11,648 KB
最終ジャッジ日時 2024-07-08 10:00:21
合計ジャッジ時間 2,886 ms
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample RE * 3
other RE * 37
権限があれば一括ダウンロードができます

ソースコード

diff #

import fractions

def read_data():
    N = int(input())
    K = int(input())
    amida = list(range(N))
    for k in range(K):
        x, y = map(int, input().split())
        amida[x-1], amida[y-1] = amida[y-1], amida[x-1]
    return N, amida

def solve(N, amida):
    cycles = [0] * N
    for idx in range(N):
        if cycles[idx]:
            continue
        find_cycle(N, amida, idx, cycles)
    uniq_cycles = set(cycles)
    lcm = 1
    for c in uniq_cycles:
        lcm = lcm * c // fractions.gcd(lcm, c)
    return lcm

def find_cycle(N, amida, idx, cycles):
    a = amida[idx]
    count = 1
    path = [idx]
    while a != idx:
        path.append(a)
        a = amida[a]
        count += 1
    for a in path:
        cycles[a] = count

N, amida = read_data()
print(solve(N, amida))
0