library_checker/data_structure/
point_set_range_frequency.rs1#[doc(no_inline)]
2pub use competitive::data_structure::RangeFrequency;
3use competitive::prelude::*;
4
5competitive::define_enum_scan! {
6 enum Query: usize {
7 0 => Set { k: usize, v: i32 }
8 1 => Query { l: usize, r: usize, x: i32 }
9 }
10}
11
12#[verify::library_checker("point_set_range_frequency")]
13pub fn point_set_range_frequency(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, a: [i32; n]);
17 let mut rf = RangeFrequency::new(a);
18 for _ in 0..q {
19 scan!(scanner, query: Query);
20 match query {
21 Query::Set { k, v } => {
22 rf.set(k, v);
23 }
24 Query::Query { l, r, x } => {
25 rf.query(l, r, x);
26 }
27 }
28 }
29 let results = rf.execute();
30 iter_print!(writer, @lf @it results);
31}