結果
問題 | No.738 平らな農地 |
ユーザー |
![]() |
提出日時 | 2020-12-10 23:47:09 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 366 ms / 2,000 ms |
コード長 | 1,802 bytes |
コンパイル時間 | 271 ms |
コンパイル使用メモリ | 82,452 KB |
実行使用メモリ | 124,256 KB |
最終ジャッジ日時 | 2024-09-19 20:50:59 |
合計ジャッジ時間 | 19,597 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 5 |
other | AC * 87 |
ソースコード
class fenwick_tree(object):def __init__(self, n):self.n = nself.log = n.bit_length()self.data = [0] * ndef __sum(self, r):s = 0while r > 0:s += self.data[r - 1]r -= r & -rreturn sdef add(self, p, x):""" a[p] += xを行う"""p += 1while p <= self.n:self.data[p - 1] += xp += p & -pdef sum(self, l, r):"""a[l] + a[l+1] + .. + a[r-1]を返す"""return self.__sum(r) - self.__sum(l)def lower_bound(self, x):"""a[0] + a[1] + .. a[i] >= x となる最小のiを返す"""if x <= 0:return -1i = 0k = 1 << self.logwhile k:if i + k <= self.n and self.data[i + k - 1] < x:x -= self.data[i + k - 1]i += kk >>= 1return idef show(self):for i in range(self.n):print(self.__sum(i+1) if not i else self.sum(i, i+1), end=" ")print()N, K = map(int, input().split())A = list(map(int, input().split()))if K == 1:print(0)exit()Aset = sorted(set(A))atoi = {a: i for i, a in enumerate(Aset)}bit_cnt = fenwick_tree(N)bit_val = fenwick_tree(N)for i in range(K - 1):a = A[i]p = atoi[a]bit_cnt.add(p, 1)bit_val.add(p, a)ans = 10 ** 18for i in range(K - 1, N):a = A[i]p = atoi[a]bit_cnt.add(p, 1)bit_val.add(p, a)mid = bit_cnt.lower_bound((K + 1) // 2)a = Aset[mid]cost = a * bit_cnt.sum(0, mid) - bit_val.sum(0, mid)cost += bit_val.sum(mid + 1, N) - a * bit_cnt.sum(mid + 1, N)ans = min(ans, cost)a = A[i - K + 1]p = atoi[a]bit_cnt.add(p, -1)bit_val.add(p, -a)print(ans)