ABC254C - K Swap

気づくまでに時間がかかったのでメモ．

考え方

$K=1$ならソートできる．したがって，$\mod K$ごとにソートできる．

その結果を$A$のソート結果と一致するか見る．

回答例

N, K = map(int, input().split())
A = list(map(int, input().split()))
A2 = sorted(A)

for i in range(K):
    tmp = sorted(A[i::K])
    for j, a in enumerate(tmp):
        if a != A2[i + j * K]:
            exit(print('No'))
print('Yes')