verification/library-checker/range_affine_range_sum/src/main.rs

• View this file on GitHub
• Last update: 2025-05-29 20:17:30+09:00
• Problem: https://judge.yosupo.jp/problem/range_affine_range_sum

Depends on

• Lazy Segment Tree (data-structure/lazy-segment-tree/src/lib.rs)
• data-structure/lazy-segment-tree/src/wrapper.rs
• Modint (math/modint/src/lib.rs)

Code

// verification-helper: PROBLEM https://judge.yosupo.jp/problem/range_affine_range_sum

use proconio::input;

use lazy_segment_tree::LazySegmentTree;
use modint::StaticModint;

type Mint = StaticModint<998244353>;

fn main() {
    input! {
        n: usize,
        q: usize,
        a: [u64; n],
    }
    let mut lst = LazySegmentTree::<(Mint, Mint), (Mint, Mint)>::new(
        n,
        |a, b| (a.0 + b.0, a.1 + b.1),
        (Mint::new(0), Mint::new(0)),
        |a, b| (a.0 * b.0 + a.1 * b.1, b.1),
        |a, b| (a.0 * b.0, a.0 * b.1 + a.1),
        (Mint::new(1), Mint::new(0)),
    );
    let a = a.iter().map(|&a| (Mint::new(a), Mint::new(1))).collect::<Vec<_>>();
    lst.build(a);
    for _ in 0..q {
        input! { query: usize, }
        match query {
            0 => {
                input! { l: usize, r: usize, b: u64, c: u64, }
                lst.apply(l, r, (Mint::new(b), Mint::new(c)));
            }
            1 => {
                input! { l: usize, r: usize, }
                println!("{}", lst.prod(l, r).0);
            }
            _ => unreachable!(),
        }
    }
}

Traceback (most recent call last):
  File "/opt/hostedtoolcache/Python/3.11.12/x64/lib/python3.11/site-packages/onlinejudge_verify/documentation/build.py", line 71, in _render_source_code_stat
    bundled_code = language.bundle(stat.path, basedir=basedir, options={'include_paths': [basedir]}).decode()
                   ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/opt/hostedtoolcache/Python/3.11.12/x64/lib/python3.11/site-packages/onlinejudge_verify/languages/rust.py", line 288, in bundle
    raise NotImplementedError
NotImplementedError

Back to top page