abc130_d
数列が与えられて、その数列の連続部分列で、総和がK以上のものを数え上げる
数列は最大10^5まで
まずは素朴なもの(といっても累積和は使ってしまったのでもっと素朴な書き方ができるが…)
pythondef solve(N, K, XS):
from itertools import accumulate
acc = [0] + list(accumulate(XS))
ret = 0
for i in range(N):
for j in range(i + 1, N + 1):
if acc[j] - acc[i] >= K:
ret += 1
return retで、これだと二乗のオーダーで10^10なので、片方をlog Nに変える
pythondef solve(N, K, XS):
import bisect
from itertools import accumulate
acc = [0] + list(accumulate(XS))
ret = 0
for i in range(N):
lb = acc[i] + K
j = bisect.bisect_left(acc, lb)
ret += N + 1 - j
return retRelated Pages
- →数え上げ×degwer×状態をまとめる×dpは全探索の高速化×arc059f×codefestival_2016_final_f×aoj2439×探索順の変更×大きい順に並べる×aoj2333×順列は挿入dp×bit_dp×区間は終点でソート×条件の言い換え×操作は多いが産物は少ない×agc013d×線形和への分解×演算順序の変更×ビット演算を桁ごとに分解×部分群×操作が可逆で全域→部分群×ラグランジュの定理×再帰的定義→dp×arc037d×桁dp×aoj0570×累積和×フェニック木×高速フーリエ変換×ntt×高速ゼータ変換×and_と_add_の畳み込み×二分累乗×agc013e×行列木定理×全域木の個数×lgv公式×非交叉経路の個数×小さい確率を無視する×二項係数の公式×経路数×45度回転×xとyにわける×カタラン数×包除原理×agc005d×約数系包除×arc064f→
"Engineer's way of creating knowledge" the English version of my book is now available on [Engineer's way of creating knowledge]
(C)NISHIO Hirokazu / Converted from [Scrapbox] at [Edit]