pub mod scanner { pub struct Scanner { buf: Vec, } impl Scanner { pub fn new() -> Self { Self { buf: vec![] } } pub fn new_from(source: &str) -> Self { let source = String::from(source); let buf = Self::split(source); Self { buf } } pub fn next(&mut self) -> T { loop { if let Some(x) = self.buf.pop() { return x.parse().ok().expect(""); } let mut source = String::new(); std::io::stdin().read_line(&mut source).expect(""); self.buf = Self::split(source); } } fn split(source: String) -> Vec { source .split_whitespace() .rev() .map(String::from) .collect::>() } } } use crate::scanner::Scanner; fn main() { let mut scanner = Scanner::new(); let t: usize = 1; for _ in 0..t { solve(&mut scanner); } } fn solve(scanner: &mut Scanner) { let _n: usize = scanner.next(); let m: usize = scanner.next(); let mut now: usize = 0; let mut a: Vec = (0..m) .map(|_| { let x = scanner.next::(); now += x * x; x }) .collect(); let q: usize = scanner.next(); for _ in 0..q { let c: usize = scanner.next::() - 1; let k: usize = scanner.next::(); let d: usize = scanner.next::() - 1; now -= a[c] * a[c]; now -= a[d] * a[d]; a[c] -= k; a[d] += k; now += a[c] * a[c]; now += a[d] * a[d]; println!("{}", now); } }