結果

問題 No.366 ロボットソート
コンテスト
ユーザー Đức Minh Vũ
提出日時 2026-02-10 17:06:43
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
WA  
実行時間 -
コード長 1,077 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 2,458 ms
コンパイル使用メモリ 82,380 KB
実行使用メモリ 57,888 KB
最終ジャッジ日時 2026-02-10 17:06:49
合計ジャッジ時間 2,650 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 15 WA * 8
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

import sys
from collections import defaultdict, deque

n, k = map(int, input().split())
a = list(map(int, input().split()))

b = sorted(a)

# lưu tất cả vị trí của từng giá trị trong b
pos = defaultdict(deque)
for i in range(n):
    pos[b[i]].append(i)

# xây permutation
perm = [-1] * n

for i in range(n):
    val = a[i]
    
    if not pos[val]:
        print(-1)
        sys.exit()
    
    # tìm vị trí phù hợp cùng mod k
    found = False
    for _ in range(len(pos[val])):
        candidate = pos[val][0]
        
        if candidate % k == i % k:
            perm[i] = candidate
            pos[val].popleft()
            found = True
            break
        else:
            pos[val].rotate(-1)
    
    if not found:
        print(-1)
        sys.exit()

# đếm cycle
visited = [False]*n
ans = 0

for i in range(n):
    if visited[i]:
        continue
    
    cycle = 0
    j = i
    
    while not visited[j]:
        visited[j] = True
        j = perm[j]
        cycle += 1
    
    if cycle > 1:
        ans += cycle - 1

print(ans)
0