結果
問題 | No.1248 Kth Sum |
ユーザー | akakimidori |
提出日時 | 2022-11-03 00:09:06 |
言語 | Rust (1.77.0 + proconio) |
結果 |
RE
|
実行時間 | - |
コード長 | 7,174 bytes |
コンパイル時間 | 13,504 ms |
コンパイル使用メモリ | 406,848 KB |
実行使用メモリ | 21,076 KB |
最終ジャッジ日時 | 2024-07-17 11:50:46 |
合計ジャッジ時間 | 18,500 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1 ms
6,816 KB |
testcase_01 | AC | 1 ms
6,812 KB |
testcase_02 | AC | 1 ms
6,944 KB |
testcase_03 | RE | - |
testcase_04 | AC | 1 ms
6,940 KB |
testcase_05 | AC | 1 ms
6,940 KB |
testcase_06 | AC | 1 ms
6,944 KB |
testcase_07 | AC | 1 ms
6,940 KB |
testcase_08 | AC | 1 ms
6,944 KB |
testcase_09 | AC | 1 ms
6,940 KB |
testcase_10 | AC | 1 ms
6,940 KB |
testcase_11 | AC | 1 ms
6,944 KB |
testcase_12 | AC | 1 ms
6,940 KB |
testcase_13 | WA | - |
testcase_14 | AC | 80 ms
19,416 KB |
testcase_15 | WA | - |
testcase_16 | AC | 92 ms
20,796 KB |
testcase_17 | AC | 90 ms
20,792 KB |
testcase_18 | AC | 87 ms
20,900 KB |
testcase_19 | AC | 90 ms
20,708 KB |
testcase_20 | AC | 90 ms
20,304 KB |
testcase_21 | AC | 88 ms
20,504 KB |
testcase_22 | AC | 77 ms
20,124 KB |
testcase_23 | AC | 1 ms
6,940 KB |
testcase_24 | AC | 94 ms
19,720 KB |
testcase_25 | AC | 115 ms
20,684 KB |
testcase_26 | WA | - |
testcase_27 | AC | 15 ms
9,452 KB |
testcase_28 | AC | 15 ms
9,332 KB |
testcase_29 | RE | - |
testcase_30 | AC | 87 ms
20,776 KB |
testcase_31 | AC | 88 ms
20,428 KB |
testcase_32 | AC | 88 ms
20,404 KB |
testcase_33 | WA | - |
testcase_34 | WA | - |
testcase_35 | WA | - |
testcase_36 | WA | - |
testcase_37 | WA | - |
ソースコード
// 部分列に分ける // 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)) 個部分列を作らないといかん // // 末尾からやればいける fn run() { input! { n: usize, k: usize, a: [u64; n], } 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); if x - 1 >= need * k - 1 { ans = ans.min(a[i] + q); } else { ans = ans.min(a[i] + q - p.0 + rmq.find(need * k - 1, n)); } } 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::<Vec<_>>() }; ($iter:expr, chars) => { read_value!($iter, String).chars().collect::<Vec<char>>() }; ($iter:expr, bytes) => { read_value!($iter, String).bytes().collect::<Vec<u8>>() }; ($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<T, F> { n: usize, size: usize, data: Vec<T>, e: T, op: F, } impl<T, F> SegmentTreePURQ<T, F> 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<P>(&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<P>(&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 ----------