結果
問題 | No.738 平らな農地 |
ユーザー | shotoyoo |
提出日時 | 2021-06-16 23:14:52 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 325 ms / 2,000 ms |
コード長 | 2,847 bytes |
コンパイル時間 | 223 ms |
コンパイル使用メモリ | 82,384 KB |
実行使用メモリ | 107,772 KB |
最終ジャッジ日時 | 2024-06-10 03:53:58 |
合計ジャッジ時間 | 19,786 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 38 ms
53,376 KB |
testcase_01 | AC | 41 ms
52,608 KB |
testcase_02 | AC | 37 ms
52,736 KB |
testcase_03 | AC | 38 ms
52,864 KB |
testcase_04 | AC | 37 ms
53,248 KB |
testcase_05 | AC | 94 ms
77,188 KB |
testcase_06 | AC | 98 ms
77,200 KB |
testcase_07 | AC | 101 ms
77,824 KB |
testcase_08 | AC | 76 ms
76,544 KB |
testcase_09 | AC | 62 ms
69,120 KB |
testcase_10 | AC | 54 ms
63,616 KB |
testcase_11 | AC | 78 ms
75,904 KB |
testcase_12 | AC | 61 ms
68,736 KB |
testcase_13 | AC | 94 ms
77,204 KB |
testcase_14 | AC | 66 ms
71,040 KB |
testcase_15 | AC | 235 ms
98,244 KB |
testcase_16 | AC | 247 ms
100,720 KB |
testcase_17 | AC | 301 ms
100,800 KB |
testcase_18 | AC | 247 ms
98,688 KB |
testcase_19 | AC | 294 ms
101,248 KB |
testcase_20 | AC | 267 ms
100,724 KB |
testcase_21 | AC | 295 ms
101,504 KB |
testcase_22 | AC | 287 ms
100,800 KB |
testcase_23 | AC | 305 ms
101,376 KB |
testcase_24 | AC | 310 ms
100,788 KB |
testcase_25 | AC | 62 ms
69,632 KB |
testcase_26 | AC | 61 ms
69,376 KB |
testcase_27 | AC | 62 ms
70,016 KB |
testcase_28 | AC | 61 ms
70,144 KB |
testcase_29 | AC | 65 ms
70,912 KB |
testcase_30 | AC | 64 ms
69,760 KB |
testcase_31 | AC | 60 ms
69,888 KB |
testcase_32 | AC | 62 ms
69,120 KB |
testcase_33 | AC | 62 ms
70,912 KB |
testcase_34 | AC | 61 ms
68,992 KB |
testcase_35 | AC | 59 ms
68,736 KB |
testcase_36 | AC | 63 ms
69,632 KB |
testcase_37 | AC | 63 ms
70,272 KB |
testcase_38 | AC | 63 ms
70,272 KB |
testcase_39 | AC | 65 ms
71,168 KB |
testcase_40 | AC | 64 ms
70,272 KB |
testcase_41 | AC | 67 ms
71,296 KB |
testcase_42 | AC | 64 ms
68,992 KB |
testcase_43 | AC | 64 ms
69,248 KB |
testcase_44 | AC | 63 ms
70,272 KB |
testcase_45 | AC | 275 ms
102,728 KB |
testcase_46 | AC | 260 ms
99,584 KB |
testcase_47 | AC | 268 ms
102,400 KB |
testcase_48 | AC | 239 ms
102,528 KB |
testcase_49 | AC | 268 ms
101,044 KB |
testcase_50 | AC | 265 ms
102,656 KB |
testcase_51 | AC | 266 ms
102,400 KB |
testcase_52 | AC | 281 ms
101,412 KB |
testcase_53 | AC | 249 ms
102,144 KB |
testcase_54 | AC | 268 ms
102,656 KB |
testcase_55 | AC | 261 ms
102,272 KB |
testcase_56 | AC | 285 ms
102,400 KB |
testcase_57 | AC | 242 ms
99,416 KB |
testcase_58 | AC | 260 ms
102,444 KB |
testcase_59 | AC | 256 ms
102,880 KB |
testcase_60 | AC | 269 ms
103,168 KB |
testcase_61 | AC | 262 ms
102,888 KB |
testcase_62 | AC | 249 ms
99,700 KB |
testcase_63 | AC | 256 ms
103,040 KB |
testcase_64 | AC | 266 ms
102,912 KB |
testcase_65 | AC | 287 ms
100,828 KB |
testcase_66 | AC | 302 ms
100,964 KB |
testcase_67 | AC | 242 ms
103,808 KB |
testcase_68 | AC | 239 ms
107,136 KB |
testcase_69 | AC | 292 ms
106,880 KB |
testcase_70 | AC | 268 ms
107,196 KB |
testcase_71 | AC | 148 ms
89,728 KB |
testcase_72 | AC | 147 ms
90,400 KB |
testcase_73 | AC | 150 ms
91,908 KB |
testcase_74 | AC | 158 ms
91,904 KB |
testcase_75 | AC | 263 ms
104,188 KB |
testcase_76 | AC | 239 ms
107,028 KB |
testcase_77 | AC | 271 ms
104,064 KB |
testcase_78 | AC | 288 ms
107,392 KB |
testcase_79 | AC | 311 ms
107,572 KB |
testcase_80 | AC | 260 ms
107,648 KB |
testcase_81 | AC | 297 ms
106,648 KB |
testcase_82 | AC | 287 ms
107,772 KB |
testcase_83 | AC | 211 ms
91,392 KB |
testcase_84 | AC | 194 ms
91,124 KB |
testcase_85 | AC | 325 ms
107,488 KB |
testcase_86 | AC | 297 ms
103,776 KB |
testcase_87 | AC | 179 ms
106,752 KB |
testcase_88 | AC | 176 ms
103,668 KB |
testcase_89 | AC | 39 ms
53,248 KB |
testcase_90 | AC | 36 ms
52,992 KB |
testcase_91 | AC | 37 ms
52,864 KB |
ソースコード
import sys input = lambda : sys.stdin.readline().rstrip() sys.setrecursionlimit(2*10**5+10) write = lambda x: sys.stdout.write(x+"\n") debug = lambda x: sys.stderr.write(x+"\n") writef = lambda x: print("{:.12f}".format(x)) ### 座標圧縮 (一次元) from bisect import bisect_left def press(l): """xs: ソートした座標値 inds: lの各要素のxsにおけるインデックス """ # xs[inds[i]]==l[i]となる xs = sorted(set(l)) inds = [None] * len(l) # d = {} for i,item in enumerate(l): inds[i] = bisect_left(xs, item) # d[item] = inds[i] return xs, inds 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 n,k = list(map(int, input().split())) a = list(map(int, input().split())) xs, inds = press(a) m = len(xs) bit = BIT(m) bit2 = BIT(m) total = 0 for i in range(k): bit.add(inds[i], 1) bit2.add(inds[i], xs[inds[i]]) total += a[i] ans = float("inf") for i in range(k,n+1): v = bit.index((k+1)//2) left = bit2.query(v) lnum = bit.query(v) right = total - left - bit2.get(v) rnum = k - lnum - bit.get(v) val = (xs[v]*lnum - left) + (right - xs[v]*rnum) ans = min(ans, val) # bit2.check() # print(left, right) if i==n: break bit.add(inds[i], 1) bit2.add(inds[i], a[i]) bit.add(inds[i-k], -1) bit2.add(inds[i-k], -a[i-k]) total += a[i] - a[i-k] print(ans)