結果

問題 No.482 あなたの名は
ユーザー rpy3cpp
提出日時 2017-03-26 01:17:27
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 185 ms / 2,000 ms
コード長 624 bytes
コンパイル時間 94 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 34,120 KB
最終ジャッジ日時 2024-07-06 06:17:44
合計ジャッジ時間 4,574 ms
ジャッジサーバーID
(参考情報)
judge2 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff #

def read_data():
    N, K = map(int, input().split())
    Ds = list(map(int, input().split()))
    return N, K, Ds

def solve(N, K, Ds):
    group = list(range(N))
    Ds = [d - 1 for d in Ds]
    n_groups = 0
    for v in range(N):
        if group[v] != v:
            continue
        n_groups += 1
        u = v
        while Ds[u] != v:
            u = Ds[u]
            group[u] = v
    min_steps = N - n_groups
    if K < min_steps:
        return False
    elif (K - min_steps) % 2:
        return False
    else:
        return True

N, K, Ds = read_data()
if solve(N, K, Ds):
    print('YES')
else:
    print('NO')
0