use std::collections::{HashSet, BTreeSet}; #[derive(Debug, Clone)] struct SegmentTree { n: usize, orgn: usize, unit: T, mem: Vec, operator: F } impl SegmentTree where T: std::fmt::Debug + Copy, F: Fn(T, T) -> T { fn new(n: usize, unit: T, operator: F) -> Self { let mut size = 1usize; while size < n { size *= 2; } Self { n: size, orgn: n, unit: unit, mem: vec![unit; size*2-1], operator: operator, } } fn find_closed(&self, l: usize, r: usize) -> T { self.find(l, r+1) } fn find(&self, l: usize, r: usize) -> T { self._find(l, r, 0, 0, self.n) } fn set(&mut self, pos: usize, val: T) { let mut pos = pos + self.n - 1; self.mem[pos] = val; while pos > 0 { pos = self._parent(pos); self.mem[pos] = (self.operator)(self.mem[self._left(pos)], self.mem[self._right(pos)]); } } fn get(&mut self, pos: usize) -> T { self.mem[pos + self.n - 1] } fn _find(&self, a: usize, b: usize, nodenum: usize, l: usize, r: usize) -> T { if r <= a || b <= l { return self.unit; } else if a <= l && r <= b { return self.mem[nodenum]; } let left = self._find(a, b, self._left(nodenum), l, (l+r)/2); let right = self._find(a, b, self._right(nodenum), (l+r)/2, r); (self.operator)(left, right) } fn _left(&self, k: usize) -> usize { 2*k+1 } fn _right(&self, k: usize) -> usize { 2*k+2 } fn _parent(&self, k: usize) -> usize { (k-1)/2 } } fn main() { let mut nq = String::new(); std::io::stdin().read_line(&mut nq).ok(); let nq: Vec = nq.trim().split_whitespace().map(|s| s.parse().unwrap()).collect(); let n = nq[0]; let q = nq[1]; let mut a = String::new(); std::io::stdin().read_line(&mut a).ok(); let mut a: Vec = a.trim().split_whitespace().map(|s| s.parse().unwrap()).collect(); let mut queries = (0..q).map(|i| { let mut temp = String::new(); std::io::stdin().read_line(&mut temp).ok(); let temp: Vec = temp.trim().split_whitespace().map(|s| s.parse().unwrap()).collect(); (temp[0]-1, temp[1], temp[2], i) }) .collect::>(); let mut a = (0..n).map(|i| (a[i], i)).collect::>(); a.sort_by_key(|&(v, _)| v); let mut result = vec![0usize; q]; let mut val_segtree = SegmentTree::new(n, 0usize, |x, y| x + y); let mut cnt_segtree = SegmentTree::new(n, 0usize, |x, y| x + y); queries.sort_by_key(|&(_, _, x, _)| x); let mut idx = 0usize; for &(l, r, x, qidx) in queries.iter() { while idx < n && a[idx].0 <= x { let (aval, aidx) = a[idx]; val_segtree.set(aidx, aval); cnt_segtree.set(aidx, 1); idx += 1; } result[qidx] = val_segtree.find(l, r) + (r - l - cnt_segtree.find(l, r)) * x; } for &v in result.iter() { println!("{}", v); } }