ABC254 C - K Swap
提出
code: python
n, k = map(int, input().split())
a = list(map(int, input().split()))
# 3 4 1 3 4
# 1 3 3 4 4
# 大抵は行ける気がする 端にあったら無理
解答
code: python
n, k = map(int, input().split())
a = list(map(int, input().split())) # 3 4 1 3 4
b = [[] for _ in range(k)]
# i を kで割った余りごとに管理
# 3, 1, 4 / 4, 3
for i, x in enumerate(a):
bi % k.append(x)
for i in range(k):
bi.sort() # ソートする
# print(b) 1, 3, 4], [3, 4
res = 0 * n
for i in range(n):
resi = bi % ki // k
print("Yes" if res == sorted(a) else "No")
メモ
【AtCoder解説】PythonでABC254のA,B,C,D,E問題を制する!