結果
問題 | No.1838 Modulo Straight |
ユーザー | shotoyoo |
提出日時 | 2022-02-11 23:09:34 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 569 ms / 2,000 ms |
コード長 | 2,406 bytes |
コンパイル時間 | 319 ms |
コンパイル使用メモリ | 87,128 KB |
実行使用メモリ | 166,388 KB |
最終ジャッジ日時 | 2023-09-10 05:40:53 |
合計ジャッジ時間 | 15,138 ms |
ジャッジサーバーID (参考情報) |
judge15 / judge14 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 71 ms
71,560 KB |
testcase_01 | AC | 69 ms
71,296 KB |
testcase_02 | AC | 69 ms
71,544 KB |
testcase_03 | AC | 90 ms
76,728 KB |
testcase_04 | AC | 87 ms
76,760 KB |
testcase_05 | AC | 87 ms
77,008 KB |
testcase_06 | AC | 84 ms
77,012 KB |
testcase_07 | AC | 88 ms
77,028 KB |
testcase_08 | AC | 90 ms
76,768 KB |
testcase_09 | AC | 92 ms
76,896 KB |
testcase_10 | AC | 92 ms
76,648 KB |
testcase_11 | AC | 569 ms
166,188 KB |
testcase_12 | AC | 558 ms
166,016 KB |
testcase_13 | AC | 559 ms
166,224 KB |
testcase_14 | AC | 554 ms
166,068 KB |
testcase_15 | AC | 552 ms
166,388 KB |
testcase_16 | AC | 493 ms
159,652 KB |
testcase_17 | AC | 493 ms
159,488 KB |
testcase_18 | AC | 498 ms
159,604 KB |
testcase_19 | AC | 460 ms
158,276 KB |
testcase_20 | AC | 449 ms
158,432 KB |
testcase_21 | AC | 444 ms
158,184 KB |
testcase_22 | AC | 278 ms
160,920 KB |
testcase_23 | AC | 282 ms
162,980 KB |
testcase_24 | AC | 280 ms
149,652 KB |
testcase_25 | AC | 274 ms
133,524 KB |
testcase_26 | AC | 287 ms
136,256 KB |
testcase_27 | AC | 297 ms
139,244 KB |
testcase_28 | AC | 309 ms
139,608 KB |
testcase_29 | AC | 314 ms
142,784 KB |
testcase_30 | AC | 337 ms
147,168 KB |
testcase_31 | AC | 374 ms
148,928 KB |
testcase_32 | AC | 402 ms
152,908 KB |
testcase_33 | AC | 491 ms
160,416 KB |
testcase_34 | AC | 472 ms
160,112 KB |
testcase_35 | AC | 408 ms
156,680 KB |
testcase_36 | AC | 385 ms
149,924 KB |
testcase_37 | AC | 332 ms
147,636 KB |
testcase_38 | AC | 307 ms
140,308 KB |
testcase_39 | AC | 165 ms
104,032 KB |
testcase_40 | AC | 69 ms
71,444 KB |
ソースコード
import sys input = lambda : sys.stdin.readline().rstrip() write = lambda x: sys.stdout.write(x+"\n"); writef = lambda x: print("{:.12f}".format(x)) debug = lambda x: sys.stderr.write(x+"\n") YES="Yes"; NO="No"; pans = lambda v: print(YES if v else NO) LI = lambda : list(map(int, input().split())) sys.setrecursionlimit(3*10**5+10) class BIT: ### BIT binary def __init__(self, n, values=None): self.bit = [0]*(n+1) self.n = n self.total = 0 if values is not None: for i,v in enumerate(values): self.add(i,v) self.total += v def check(self): l = [] prv = 0 for i in range(1,self.n+1): val = self.query(i) l.append(val-prv) prv = val print(" ".join(map(str, l))) #a1 ~ aiまでの和 O(logn) def query(self,i): res = 0 while i > 0: res += self.bit[i] # res %= M i -= i&(-i) return res def get(self,i): return self.query(i+1) - self.query(i) #ai += x(logN) def add(self,i,x): i += 1 if i==0: raise RuntimeError self.total += x while i <= self.n: self.bit[i] += x # self.bit[i] %= M i += i&(-i) def index(self, v): """a0,...,aiの和がv以上になる最小のindexを求める 存在しないとき配列サイズを返す """ if v <= 0: return 0 if self.total<v: return self.n x = 0 r = 1 while r < self.n: r = r << 1; ll = r while ll>0: if x+ll<self.n and self.bit[x+ll]<v: v -= self.bit[x+ll] x += ll ll = ll>>1 return x m,k = LI() a = list(map(int, input().split())) n = len(a) bit = [0]*k count = [0]*m index = [[] for _ in range(m)] vals = [0]*n vs = [0]*n for i in range(n): v = a[i] vs[i] = v+m*count[v] index[v].append(i) val = bit[count[v]] bit[count[v]] += 1 vals[i] = val count[v] += 1 bit2 = BIT(n) total = 0 inv = 0 for v in vs: inv += total - bit2.query(v) bit2.add(v, 1) total += 1 ans = inv for i in range(m): num = 0 for ind in index[i]: num += vals[ind] inv += (-num) + ((m-1)*k-num) ans = min(ans, inv) print(ans)