use std::cmp::Reverse; use proconio::{input, marker::Usize1}; fn main() { input! { n: usize, q: usize, a: [usize; n], queries: [(Usize1, usize, usize); q] } let mut cum = vec![0; n + 1]; for (i, &a) in a.iter().enumerate() { cum[i + 1] = cum[i] + a; } let mut a = a.into_iter().enumerate().collect::>(); a.sort_unstable_by_key(|&(_, a)| Reverse(a)); let mut queries = queries.into_iter().enumerate().collect::>(); queries.sort_unstable_by_key(|&(_, (_, _, x))| x); let mut ans = vec!["".to_string(); q]; let mut bit_cnt = BinaryIndexedTree::new(n, 0); let mut bit_val = BinaryIndexedTree::new(n, 0); for (i, (l, r, x)) in queries { while a.last().map_or(false, |&(_, a)| a < x) { let (j, a) = a.pop().unwrap(); bit_cnt.add(j, 1); bit_val.add(j, a); } let mut res = (cum[r] - cum[l]) as isize - x as isize * (r - l) as isize; let cnt = bit_cnt.sum(l, r) as isize; let val = bit_val.sum(l, r) as isize; res += 2 * (cnt * x as isize - val); ans[i] = res.to_string(); } println!("{}", ans.join("\n")); } pub struct BinaryIndexedTree { n: usize, array: Vec, e: T, } impl BinaryIndexedTree where T: Clone + std::ops::AddAssign + std::ops::Sub + std::cmp::PartialOrd, { pub fn new(n: usize, e: T) -> Self { Self { n, array: vec![e.clone(); n + 1], e, } } pub fn add(&mut self, i: usize, x: T) { let mut i = i + 1; while i <= self.n { self.array[i] += x.clone(); i += i & i.wrapping_neg(); } } pub fn cum(&self, mut i: usize) -> T { let mut cum = self.e.clone(); while i > 0 { cum += self.array[i].clone(); i -= i & i.wrapping_neg(); } cum } pub fn sum(&self, l: usize, r: usize) -> T { self.cum(r) - self.cum(l) } pub fn lower_bound(&self, mut x: T) -> usize { let mut i = 0; let mut k = 1 << (self.n.next_power_of_two().trailing_zeros()); while k > 0 { if i + k <= self.n && self.array[i + k] < x { x = x - self.array[i + k].clone(); i += k; } k >>= 1; } i } }