// 部分列に分ける // 1-indexed として // 1個の時はa_k しかない // 2個に分ける時 // どういうものが作れるか // [k, 2k-1] から一つは取らないといけない // とった場所をxとして // もう片方はmax(x + 1, 2k)以降から取れる // 3個の時は? // [k-1+1, 3(k-1)+1]から一つ // とった場所をxとして // max(2k, x + 1)以降から一つ, yとして // max(3k, y+1) から一つ取れる // ... // 最初に取る場所決めると取らないといけない個数が決まる // 取る個数余計に増やしても嬉しくならないので // その範囲で最小化したい // 最初の要素決めたらそれ以降の要素は好きに取れる? // 1個だけ条件ついてる // 上位i個を管理しつつ添字maxが条件満たしてるならそのまま // 満たしてないなら消して最小値を追加 // みたいなことやればいい // // 最初にとった要素がxとして // b = 1 + ceil((x-k)/(k-1)) 個部分列を作らないといかん // // 末尾からやればいける // // WA // needがミスってる fn run() { input! { n: usize, k: usize, a: [u64; n], } if k == 1 { println!("{}", a[0]); return; } let mut ans = a[k - 1]; let mut seg = SegmentTreePURQ::new(n, ((0, 0), 0), |a, b| (std::cmp::max(a.0, b.0), a.1 + b.1)); let mut rmq = SegmentTreePURQ::new(n, std::u64::MAX / 3, |a, b| std::cmp::min(*a, *b)); for i in 0..n { rmq.update_tmp(i, a[i]); } rmq.update_all(); let mut cnt = 0; for i in (k..n).rev() { let need = (i + 1 - k + k - 2) / (k - 1); while cnt > need { let ((_, x), _) = seg.find(0, n); seg.update(x, ((0, 0), 0)); cnt -= 1; } if cnt == need { let x = seg.min_left(n, |p| p.1 == 0); let (p, q) = seg.find(0, n); let v = if x - 1 >= (need + 1) * k - 1 { a[i] + q } else { a[i] + q - p.0 + rmq.find((need + 1) * k - 1, n) }; ans = ans.min(v); } cnt += 1; seg.update(i, ((a[i], i), a[i])); } println!("{}", ans); } fn main() { run(); } // ---------- begin input macro ---------- // reference: https://qiita.com/tanakh/items/0ba42c7ca36cd29d0ac8 #[macro_export] macro_rules! input { (source = $s:expr, $($r:tt)*) => { let mut iter = $s.split_whitespace(); input_inner!{iter, $($r)*} }; ($($r:tt)*) => { let s = { use std::io::Read; let mut s = String::new(); std::io::stdin().read_to_string(&mut s).unwrap(); s }; let mut iter = s.split_whitespace(); input_inner!{iter, $($r)*} }; } #[macro_export] macro_rules! input_inner { ($iter:expr) => {}; ($iter:expr, ) => {}; ($iter:expr, $var:ident : $t:tt $($r:tt)*) => { let $var = read_value!($iter, $t); input_inner!{$iter $($r)*} }; } #[macro_export] macro_rules! read_value { ($iter:expr, ( $($t:tt),* )) => { ( $(read_value!($iter, $t)),* ) }; ($iter:expr, [ $t:tt ; $len:expr ]) => { (0..$len).map(|_| read_value!($iter, $t)).collect::>() }; ($iter:expr, chars) => { read_value!($iter, String).chars().collect::>() }; ($iter:expr, bytes) => { read_value!($iter, String).bytes().collect::>() }; ($iter:expr, usize1) => { read_value!($iter, usize) - 1 }; ($iter:expr, $t:ty) => { $iter.next().unwrap().parse::<$t>().expect("Parse error") }; } // ---------- end input macro ---------- // ---------- begin segment tree Point Update Range Query ---------- pub struct SegmentTreePURQ { n: usize, size: usize, data: Vec, e: T, op: F, } impl SegmentTreePURQ where T: Clone, F: Fn(&T, &T) -> T, { pub fn new(n: usize, e: T, op: F) -> Self { assert!(n > 0); let size = n.next_power_of_two(); let data = vec![e.clone(); 2 * size]; SegmentTreePURQ { n, size, data, e, op, } } pub fn update_tmp(&mut self, x: usize, v: T) { assert!(x < self.n); self.data[x + self.size] = v; } pub fn update_all(&mut self) { for i in (1..self.size).rev() { self.data[i] = (self.op)(&self.data[2 * i], &self.data[2 * i + 1]); } } pub fn update(&mut self, x: usize, v: T) { assert!(x < self.n); let mut x = x + self.size; self.data[x] = v; x >>= 1; while x > 0 { self.data[x] = (self.op)(&self.data[2 * x], &self.data[2 * x + 1]); x >>= 1; } } pub fn find(&self, l: usize, r: usize) -> T { assert!(l <= r && r <= self.n); if l == r { return self.e.clone(); } let mut l = self.size + l; let mut r = self.size + r; let mut x = self.e.clone(); let mut y = self.e.clone(); while l < r { if l & 1 == 1 { x = (self.op)(&x, &self.data[l]); l += 1; } if r & 1 == 1 { r -= 1; y = (self.op)(&self.data[r], &y); } l >>= 1; r >>= 1; } (self.op)(&x, &y) } pub fn max_right

(&self, l: usize, f: P) -> usize where P: Fn(&T) -> bool, { assert!(l <= self.n); assert!(f(&self.e)); if l == self.n { return self.n; } let mut l = l + self.size; let mut sum = self.e.clone(); while { l >>= l.trailing_zeros(); let v = (self.op)(&sum, &self.data[l]); if !f(&v) { while l < self.size { l <<= 1; let v = (self.op)(&sum, &self.data[l]); if f(&v) { sum = v; l += 1; } } return l - self.size; } sum = v; l += 1; l.count_ones() > 1 } {} self.n } pub fn min_left

(&self, r: usize, f: P) -> usize where P: Fn(&T) -> bool, { assert!(r <= self.n); assert!(f(&self.e)); if r == 0 { return 0; } let mut r = r + self.size; let mut sum = self.e.clone(); while { r -= 1; while r > 1 && r & 1 == 1 { r >>= 1; } let v = (self.op)(&self.data[r], &sum); if !f(&v) { while r < self.size { r = 2 * r + 1; let v = (self.op)(&self.data[r], &sum); if f(&v) { sum = v; r -= 1; } } return r + 1 - self.size; } sum = v; (r & (!r + 1)) != r } {} 0 } } // ---------- end segment tree Point Update Range Query ----------