Skip to main content

library_checker/data_structure/
point_set_range_frequency.rs

1use competitive::data_structure::RangeFrequency;
2use competitive::prelude::*;
3
4competitive::define_enum_scan! {
5    enum Query: usize {
6        0 => Set { k: usize, v: i32 }
7        1 => Query { l: usize, r: usize, x: i32 }
8    }
9}
10
11#[verify::library_checker("point_set_range_frequency")]
12pub fn point_set_range_frequency(reader: impl Read, writer: impl Write) {
13    prepare_io!(reader, writer);
14    sc!(n, q, a: [i32; n]);
15    let mut rf = RangeFrequency::new(a);
16    for _ in 0..q {
17        sc!(query: Query);
18        match query {
19            Query::Set { k, v } => {
20                rf.set(k, v);
21            }
22            Query::Query { l, r, x } => {
23                rf.query(l, r, x);
24            }
25        }
26    }
27    let results = rf.execute();
28    pp!(@lf @it results);
29}