結果

問題 No.2065 Sum of Min
ユーザー flygonflygon
提出日時 2022-09-19 01:37:42
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,055 ms / 2,000 ms
コード長 1,091 bytes
コンパイル時間 397 ms
コンパイル使用メモリ 82,112 KB
実行使用メモリ 176,200 KB
最終ジャッジ日時 2024-06-01 15:57:29
合計ジャッジ時間 19,183 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 45 ms
69,548 KB
testcase_01 AC 46 ms
70,264 KB
testcase_02 AC 47 ms
70,348 KB
testcase_03 AC 47 ms
70,768 KB
testcase_04 AC 659 ms
173,908 KB
testcase_05 AC 639 ms
174,084 KB
testcase_06 AC 840 ms
132,112 KB
testcase_07 AC 393 ms
120,644 KB
testcase_08 AC 736 ms
129,404 KB
testcase_09 AC 932 ms
175,616 KB
testcase_10 AC 768 ms
171,712 KB
testcase_11 AC 760 ms
171,536 KB
testcase_12 AC 1,049 ms
175,944 KB
testcase_13 AC 1,020 ms
175,708 KB
testcase_14 AC 1,055 ms
176,088 KB
testcase_15 AC 1,050 ms
175,388 KB
testcase_16 AC 1,032 ms
176,200 KB
testcase_17 AC 1,037 ms
176,080 KB
testcase_18 AC 1,031 ms
175,808 KB
testcase_19 AC 1,045 ms
175,492 KB
testcase_20 AC 1,011 ms
175,672 KB
testcase_21 AC 1,038 ms
175,988 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

n,q = map(int,input().split())
a = list(map(int,input().split()))
qs = []
for i in range(q):
  l,r,x = map(int,input().split())
  qs.append((l-1,x,i,-1))
  qs.append((r,x,i,1))
qs.sort(key = lambda x:x[0])
def compress(arr):
  *s, = set(arr)
  return {num: i for i,num in enumerate(sorted(s))}
ca = compress(a)
sa = sorted(set(a))
seg_len = 1<<19 # > 5*(10**5)
seg = [0]*(2*seg_len)
seg2 = [0]*(2*seg_len)


def update(pos,x,seg):
  pos += seg_len
  seg[pos] += x 
  while True: 
    pos >>= 1 
    if not pos: break 
    seg[pos] = seg[pos<<1] + seg[pos<<1|1]

def get_query(l,r,seg):
  l += seg_len; r += seg_len
  res = 0 
  while l < r: 
    if l & 1: 
      res += seg[l] 
      l += 1 
    if r & 1: 
      r -= 1 
      res += seg[r] 
    l >>= 1; r >>= 1
  return res
now = 0
ans =[0]*q
from bisect import bisect_left, bisect_right
for l,x,i,ki in qs:
  while now < l:
    update(ca[a[now]], 1, seg)
    update(ca[a[now]], a[now], seg2)
    now += 1
  xx = bisect_right(sa,x)
  s = get_query(0, xx, seg2)
  t = get_query(0, xx, seg)
  ans[i] += ki*(s+(l-t)*x)
print(*ans, sep = "\n")
0