結果

問題 No.366 ロボットソート
ユーザー vorg101
提出日時 2016-05-11 01:45:42
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 165 ms / 2,000 ms
コード長 1,027 bytes
コンパイル時間 167 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 10,880 KB
最終ジャッジ日時 2024-12-29 17:54:35
合計ジャッジ時間 1,842 ms
ジャッジサーバーID
(参考情報)
judge4 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 23
権限があれば一括ダウンロードができます

ソースコード

diff #

def sort_counter(lis):
    count = 0
    while True:
        innner_counter = 0
        for i in range(len(lis) - 1):
            if lis[i] > lis[i + 1]:
                lis[i], lis[i + 1] = lis[i + 1], lis[i]
                innner_counter += 1
        if innner_counter == 0:
            break
        else:
            count += innner_counter
    return count

N, K = map(int, input().split(' '))
w = list(map(int, input().split(' ')))
g = [[] for i in range(K)]
for i, n in enumerate(w):
    g[i % K].append(n)
flag = True

if w != sorted(w):
    for n in zip(*list(map(lambda l: sorted(l), g))):
        tn = n[0]
        for ttn in n[1:]:
            if tn <= ttn:
                tn = ttn
                continue
            flag = False
            break
        if flag == False:
            break

    if flag:
        count = 0
        for l in g:
            count += sort_counter(l)
        if count != 0:
            print(count)
        else:
            print(-1)
    else:
        print(-1)
else:
    print(0)
0