aizu_online_judge/dsl/
dsl_2_g.rs

1use competitive::prelude::*;
2#[doc(no_inline)]
3pub use competitive::{algebra::RangeSumRangeAdd, data_structure::LazySegmentTree};
4
5competitive::define_enum_scan! {
6    enum Query: usize {
7        0 => Add { s: usize, t: usize, x: u64 }
8        1 => Sum { s: usize, t: usize }
9    }
10}
11
12#[verify::aizu_online_judge("DSL_2_G")]
13pub fn dsl_2_g(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::<RangeSumRangeAdd<_>>::from_vec(vec![(0, 1); n]);
18    for _ in 0..q {
19        scan!(scanner, query: Query);
20        match query {
21            Query::Add { s, t, x } => {
22                seg.update(s - 1..t, x);
23            }
24            Query::Sum { s, t } => {
25                writeln!(writer, "{}", seg.fold(s - 1..t).0).ok();
26            }
27        }
28    }
29}
aizu_online_judge/dsl/dsl_2_g.rs

aizu_online_judge/dsl/
dsl_2_g.rs
