結果
問題 | No.2065 Sum of Min |
ユーザー | Navier_Boltzmann |
提出日時 | 2022-09-03 20:44:25 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,361 ms / 2,000 ms |
コード長 | 2,240 bytes |
コンパイル時間 | 286 ms |
コンパイル使用メモリ | 81,976 KB |
実行使用メモリ | 142,456 KB |
最終ジャッジ日時 | 2024-04-28 17:45:22 |
合計ジャッジ時間 | 24,087 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 48 ms
57,080 KB |
testcase_01 | AC | 48 ms
56,792 KB |
testcase_02 | AC | 47 ms
56,240 KB |
testcase_03 | AC | 49 ms
56,324 KB |
testcase_04 | AC | 828 ms
140,232 KB |
testcase_05 | AC | 776 ms
138,164 KB |
testcase_06 | AC | 1,054 ms
138,144 KB |
testcase_07 | AC | 519 ms
109,776 KB |
testcase_08 | AC | 1,028 ms
137,612 KB |
testcase_09 | AC | 1,053 ms
138,900 KB |
testcase_10 | AC | 1,030 ms
138,920 KB |
testcase_11 | AC | 1,075 ms
138,160 KB |
testcase_12 | AC | 1,361 ms
141,292 KB |
testcase_13 | AC | 1,272 ms
139,384 KB |
testcase_14 | AC | 1,183 ms
141,292 KB |
testcase_15 | AC | 1,290 ms
140,652 KB |
testcase_16 | AC | 1,218 ms
141,048 KB |
testcase_17 | AC | 1,198 ms
141,304 KB |
testcase_18 | AC | 1,169 ms
139,708 KB |
testcase_19 | AC | 1,187 ms
139,872 KB |
testcase_20 | AC | 1,195 ms
142,456 KB |
testcase_21 | AC | 1,198 ms
140,652 KB |
ソースコード
from collections import * from itertools import * from functools import * from heapq import * import sys,math input = sys.stdin.readline N,Q = map(int,input().split()) A = list(map(int,input().split())) class SegTree: """ Segment Tree """ def __init__(self, init_val, segfunc, ide_ele): """ 初期化 init_val: 配列の初期値 """ n = len(init_val) self.segfunc = segfunc self.ide_ele = ide_ele self.num = 1 << (n - 1).bit_length() self.tree = [ide_ele] * 2 * self.num # 配列の値を葉にセット for i in range(n): self.tree[self.num + i] = init_val[i] # 構築していく for i in range(self.num - 1, 0, -1): self.tree[i] = segfunc(self.tree[2 * i], self.tree[2 * i + 1]) def update(self, k, x): """ k番目の値をxに更新 k: index(0-index) x: update value """ k += self.num self.tree[k] = x while k > 1: self.tree[k >> 1] = self.segfunc(self.tree[k], self.tree[k ^ 1]) k >>= 1 def query(self, l, r): """ [l, r)のsegfuncしたものを得る l: index(0-index) r: index(0-index) """ res = self.ide_ele l += self.num r += self.num while l < r: if l & 1: res = self.segfunc(res, self.tree[l]) l += 1 if r & 1: res = self.segfunc(res, self.tree[r - 1]) l >>= 1 r >>= 1 return res S = list(accumulate([0]+A)) query = [tuple(map(int,input().split())) for _ in range(Q)] T = SegTree([(A[i],i) for i in range(N)],max,(-1,-1)) U = SegTree([0]*N,lambda x,y:x+y,0) V = SegTree([0]*N,lambda x,y:x+y,0) X = [(x,l,r) for l,r,x in query] X.sort(reverse=True) ans = defaultdict(int) for i in range(Q): x,l,r = X[i] t = (l,r,x) while T.query(0,N)[0] >= x: j = T.query(0,N)[1] V.update(j,A[j]) U.update(j,1) T.update(j,(-1,-1)) l -= 1 tans = S[r]-S[l] - V.query(l,r) + x*U.query(l,r) ans[t] = tans for q in query: print(ans[q])