結果
問題 | No.366 ロボットソート |
ユーザー | Navier_Boltzmann |
提出日時 | 2024-04-06 20:46:31 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 67 ms / 2,000 ms |
コード長 | 1,166 bytes |
コンパイル時間 | 232 ms |
コンパイル使用メモリ | 82,164 KB |
実行使用メモリ | 71,296 KB |
最終ジャッジ日時 | 2024-10-01 04:03:50 |
合計ジャッジ時間 | 2,358 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 51 ms
57,872 KB |
testcase_01 | AC | 49 ms
55,876 KB |
testcase_02 | AC | 47 ms
56,300 KB |
testcase_03 | AC | 47 ms
56,168 KB |
testcase_04 | AC | 48 ms
56,760 KB |
testcase_05 | AC | 46 ms
56,816 KB |
testcase_06 | AC | 48 ms
56,288 KB |
testcase_07 | AC | 50 ms
55,708 KB |
testcase_08 | AC | 48 ms
56,488 KB |
testcase_09 | AC | 47 ms
55,948 KB |
testcase_10 | AC | 48 ms
56,200 KB |
testcase_11 | AC | 47 ms
57,052 KB |
testcase_12 | AC | 48 ms
56,792 KB |
testcase_13 | AC | 48 ms
56,832 KB |
testcase_14 | AC | 50 ms
56,924 KB |
testcase_15 | AC | 49 ms
56,476 KB |
testcase_16 | AC | 54 ms
63,592 KB |
testcase_17 | AC | 57 ms
66,156 KB |
testcase_18 | AC | 67 ms
71,296 KB |
testcase_19 | AC | 59 ms
66,456 KB |
testcase_20 | AC | 48 ms
56,228 KB |
testcase_21 | AC | 59 ms
67,200 KB |
testcase_22 | AC | 59 ms
66,356 KB |
ソースコード
from collections import * from itertools import * from functools import * from heapq import * import sys,math input = sys.stdin.readline def mergecount(A): cnt = 0 n = len(A) if n>1: A1 = A[:n>>1] A2 = A[n>>1:] cnt += mergecount(A1) cnt += mergecount(A2) i1=0 i2=0 for i in range(n): if i2 == len(A2): A[i] = A1[i1] i1 += 1 elif i1 == len(A1): A[i] = A2[i2] i2 += 1 elif A1[i1] <= A2[i2]: A[i] = A1[i1] i1 += 1 else: A[i] = A2[i2] i2 += 1 cnt += n//2 - i1 return cnt N,K = map(int,input().split()) A = list(map(int,input().split())) B = [[] for _ in range(K)] C = [[] for _ in range(K)] for i in range(N): idx = i%K B[idx].append(A[i]) C[idx].append(A[i]) for i in range(K): C[i].sort() T = [0]*N for i in range(N): p,r = divmod(i,K) T[i]=C[r][p] if any(T[i]>=T[i+1] for i in range(N-1)): print(-1) exit() ans = 0 for b in B: ans += mergecount(b) print(ans)