結果
問題 | No.366 ロボットソート |
ユーザー | FromBooska |
提出日時 | 2023-09-04 00:22:09 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 2,688 bytes |
コンパイル時間 | 394 ms |
コンパイル使用メモリ | 81,664 KB |
実行使用メモリ | 62,720 KB |
最終ジャッジ日時 | 2024-06-13 04:51:36 |
合計ジャッジ時間 | 2,465 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 36 ms
52,096 KB |
testcase_01 | AC | 37 ms
52,224 KB |
testcase_02 | AC | 36 ms
52,096 KB |
testcase_03 | AC | 36 ms
52,352 KB |
testcase_04 | AC | 40 ms
52,224 KB |
testcase_05 | AC | 44 ms
52,224 KB |
testcase_06 | WA | - |
testcase_07 | AC | 43 ms
52,224 KB |
testcase_08 | WA | - |
testcase_09 | AC | 43 ms
52,608 KB |
testcase_10 | AC | 42 ms
52,480 KB |
testcase_11 | AC | 47 ms
54,144 KB |
testcase_12 | AC | 49 ms
54,016 KB |
testcase_13 | AC | 56 ms
60,032 KB |
testcase_14 | AC | 49 ms
54,400 KB |
testcase_15 | AC | 46 ms
53,120 KB |
testcase_16 | AC | 54 ms
60,160 KB |
testcase_17 | AC | 55 ms
61,056 KB |
testcase_18 | AC | 58 ms
62,720 KB |
testcase_19 | AC | 52 ms
60,544 KB |
testcase_20 | WA | - |
testcase_21 | AC | 52 ms
61,568 KB |
testcase_22 | AC | 49 ms
60,800 KB |
ソースコード
# まず要素を大きさ順で書き換える # mod Kで要素をグループ分け、そのグループ内なら動かせる # それぞれのグループにあるべき数字がなければ-1 # あるべき数字があれば、バブルソートの回数は転倒数と同じ # https://ikatakos.com/pot/programming_algorithm/dynamic_programming/inversion # その和が答え # リスト長が1以下なら転倒数数えなくていい # 転倒数はBinary Indexed Tree BITというデータ構造で高速に求まる # 数列の最大値以上のサイズのBITを用意、最大値がわかっている必要がある # https://scrapbox.io/pocala-kyopro/%E8%BB%A2%E5%80%92%E6%95%B0 # https://nagiss.hateblo.jp/entry/2019/07/01/185421 class Bit: def __init__(self, n): self.size = n self.tree = [0]*(n+1) def __iter__(self): psum = 0 for i in range(self.size): csum = self.sum(i+1) yield csum - psum psum = csum raise StopIteration() def __str__(self): # O(nlogn) return str(list(self)) def sum(self, i): # [0, i) の要素の総和を返す if not (0 <= i <= self.size): raise ValueError("error!") s = 0 while i>0: s += self.tree[i] i -= i & -i return s def add(self, i, x): if not (0 <= i < self.size): raise ValueError("error!") i += 1 while i <= self.size: self.tree[i] += x i += i & -i def __getitem__(self, key): if not (0 <= key < self.size): raise IndexError("error!") return self.sum(key+1) - self.sum(key) def __setitem__(self, key, value): # 足し算と引き算にはaddを使うべき if not (0 <= key < self.size): raise IndexError("error!") self.add(key, value - self[key]) def inv_count(LIST): bit = Bit(max(LIST)+1) count = 0 for i, a in enumerate(LIST): count += i - bit.sum(a+1) bit.add(a, 1) return count N, K = map(int, input().split()) A = list(map(int, input().split())) A_idx = [] for i in range(N): A_idx.append((A[i], i)) A_idx.sort() order = [-1]*N for j in range(N): a, i = A_idx[j] order[i] = j+1 groups = [[] for i in range(K)] for i in range(N): groups[i%K].append(order[i]) test = True for i in range(K): temp = sorted(groups[i]) for j in range(1, len(temp)): if temp[j]-temp[j-1] != K: test = False #print(groups) #print(test) if test == False: ans = -1 else: ans = 0 for i in range(K): if len(groups[i]) > 1: ans += inv_count(groups[i]) print(ans)