rustgym/leetcode/
_1144_decrease_elements_to_make_array_zigzag.rs

1struct Solution;
2
3impl Solution {
4    fn moves_to_make_zigzag(nums: Vec<i32>) -> i32 {
5        let mut sums = vec![0, 0];
6        let n = nums.len();
7        for i in 0..n {
8            let mut adj = vec![];
9            if i > 0 {
10                adj.push(nums[i - 1]);
11            }
12            if i + 1 < n {
13                adj.push(nums[i + 1]);
14            }
15            let min = adj.into_iter().min().unwrap();
16            if nums[i] >= min {
17                sums[i % 2] += (nums[i] - min) + 1;
18            }
19        }
20        sums[0].min(sums[1])
21    }
22}
23
24#[test]
25fn test() {
26    let nums = vec![1, 2, 3];
27    let res = 2;
28    assert_eq!(Solution::moves_to_make_zigzag(nums), res);
29    let nums = vec![9, 6, 1, 6, 2];
30    let res = 4;
31    assert_eq!(Solution::moves_to_make_zigzag(nums), res);
32    let nums = vec![10, 4, 4, 10, 10, 6, 2, 3];
33    let res = 13;
34    assert_eq!(Solution::moves_to_make_zigzag(nums), res);
35}