結果
問題 |
No.2758 RDQ
|
ユーザー |
![]() |
提出日時 | 2024-05-17 22:18:46 |
言語 | Rust (1.83.0 + proconio) |
結果 |
AC
|
実行時間 | 1,669 ms / 2,000 ms |
コード長 | 3,078 bytes |
コンパイル時間 | 19,052 ms |
コンパイル使用メモリ | 386,864 KB |
実行使用メモリ | 115,692 KB |
最終ジャッジ日時 | 2024-12-20 15:47:11 |
合計ジャッジ時間 | 35,641 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 21 |
ソースコード
#![allow(non_snake_case, unused_imports)] use proconio::{fastout, input, marker::*}; #[fastout] fn main() { input! { N: usize, Q: usize, A: [usize; N], } let mut bits = vec![DynamicBinaryIndexedTree::new(N); 100_010]; for i in 0..N { let a = A[i]; let divs = calc_divisors(a); for d in divs { bits[d].add(i, 1); } } for _ in 0..Q { input! { L: Usize1, R: Usize1, K: usize, } let ans = bits[K].sum(L..=R); println!("{}", ans); } } pub fn calc_divisors(n: usize) -> Vec<usize> { let mut ret = vec![]; for i in 1..=n { if i * i > n { break; } if n % i != 0 { continue; } ret.push(i); if n / i != i { ret.push(n / i); } } ret.sort(); return ret; } #[derive(Clone)] pub struct DynamicBinaryIndexedTree<T> { size: usize, tree: std::collections::HashMap<usize, T>, } impl< T: Default + Clone + Copy + std::ops::Add<Output = T> + std::ops::AddAssign + std::ops::Sub<Output = T>, > DynamicBinaryIndexedTree<T> { /// self = [0; size] pub fn new(size: usize) -> Self { return Self { size: size, tree: std::collections::HashMap::new(), }; } fn _inner_add(&mut self, mut i: usize, w: T) { assert!(i <= self.size && i > 0); while i <= self.size { let prev = self.tree.get(&i); match prev { Some(&v) => { self.tree.insert(i, v + w); } None => { self.tree.insert(i, w); } } i += i & i.wrapping_neg(); } } fn _inner_sum(&self, mut i: usize) -> T { let mut ret = T::default(); while i > 0 { if let Some(&v) = self.tree.get(&i) { ret += v; } i -= i & i.wrapping_neg(); } return ret; } /// self[i] += w pub fn add(&mut self, i: usize, w: T) { self._inner_add(i + 1, w); } /// return Σ_{j ∈ [0, i]} self[j] pub fn prefix_sum(&self, i: usize) -> T { self._inner_sum(i + 1) } /// return Σ_{j ∈ range} self[j] pub fn sum<R: std::ops::RangeBounds<usize>>(&self, range: R) -> T { let left = match range.start_bound() { std::ops::Bound::Included(&l) => l, std::ops::Bound::Excluded(&l) => l + 1, std::ops::Bound::Unbounded => 0, }; let right = match range.end_bound() { std::ops::Bound::Included(&r) => r, std::ops::Bound::Excluded(&r) => r - 1, std::ops::Bound::Unbounded => self.tree.len() - 1, }; if left == 0 { return self.prefix_sum(right); } else { return self.prefix_sum(right) - self.prefix_sum(left - 1); } } }