結果
問題 | No.1838 Modulo Straight |
ユーザー |
|
提出日時 | 2022-02-11 23:09:34 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 547 ms / 2,000 ms |
コード長 | 2,406 bytes |
コンパイル時間 | 191 ms |
コンパイル使用メモリ | 82,160 KB |
実行使用メモリ | 164,176 KB |
最終ジャッジ日時 | 2024-06-27 21:15:02 |
合計ジャッジ時間 | 13,474 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 38 |
ソースコード
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)