aizu_online_judge/dsl/
dsl_2_i.rs

1use competitive::prelude::*;
2#[doc(no_inline)]
3pub use competitive::{algebra::RangeSumRangeUpdate, data_structure::LazySegmentTree};
4
5competitive::define_enum_scan! {
6    enum Query: usize {
7        0 => Update { s: usize, t: usize, x: i64 }
8        1 => Sum { s: usize, t: usize }
9    }
10}
11
12#[verify::aizu_online_judge("DSL_2_I")]
13pub fn dsl_2_i(reader: impl Read, mut writer: impl Write) {
14    let s = read_all_unchecked(reader);
15    let mut scanner = Scanner::new(&s);
16    scan!(scanner, n, q);
17    let mut seg = LazySegmentTree::<RangeSumRangeUpdate<_>>::from_vec(vec![(0, 1); n]);
18    for _ in 0..q {
19        scan!(scanner, query: Query);
20        match query {
21            Query::Update { s, t, x } => {
22                seg.update(s..t + 1, Some(x));
23            }
24            Query::Sum { s, t } => {
25                writeln!(writer, "{}", seg.fold(s..t + 1).0).ok();
26            }
27        }
28    }
29}
aizu_online_judge/dsl/dsl_2_i.rs

aizu_online_judge/dsl/
dsl_2_i.rs
