use std::io::{self, Read};
fn main() {
let mut input = String::new();
io::stdin().read_to_string(&mut input).unwrap();
let mut it = input.split_whitespace();
let n: usize = it.next().unwrap().parse().unwrap();
let k: i32 = it.next().unwrap().parse().unwrap();
let mut aa: Vec<i32> = Vec::new();
let mut bb: Vec<i32> = Vec::new();
for _ in 0..n {
let ai: i32 = it.next().unwrap().parse().unwrap();
aa.push(ai);
bb.push(ai);
}
for i in 0..n {
for j in 0..n {
bb[i] = bb[i].max(aa[j] - k * (i as i32 - j as i32).abs());
}
}
print!("{}", bb.iter().zip(&aa).map(|(b, a)| (b - a) as i64).sum::<i64>());
}
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 | use std::io::{self, Read}; fn main() { let mut input = String::new(); io::stdin().read_to_string(&mut input).unwrap(); let mut it = input.split_whitespace(); let n: usize = it.next().unwrap().parse().unwrap(); let k: i32 = it.next().unwrap().parse().unwrap(); let mut aa: Vec<i32> = Vec::new(); let mut bb: Vec<i32> = Vec::new(); for _ in 0..n { let ai: i32 = it.next().unwrap().parse().unwrap(); aa.push(ai); bb.push(ai); } for i in 0..n { for j in 0..n { bb[i] = bb[i].max(aa[j] - k * (i as i32 - j as i32).abs()); } } print!("{}", bb.iter().zip(&aa).map(|(b, a)| (b - a) as i64).sum::<i64>()); } |
English