1144. 递减元素使数组呈锯齿状 (Medium)

问题描述

给你一个整数数组 nums，每次操作会从中选择一个元素并 将该元素的值减少 1。

如果符合下列情况之一，则数组 A 就是 锯齿数组：

每个偶数索引对应的元素都大于相邻的元素，即 A[0] > A[1] < A[2] > A[3] < A[4] > ...
或者，每个奇数索引对应的元素都大于相邻的元素，即 A[0] < A[1] > A[2] < A[3] > A[4] < ...

返回将数组 nums 转换为锯齿数组所需的最小操作次数。

示例 1：

输入：nums = [1,2,3]

输出：2

解释：我们可以把 2 递减到 0，或把 3 递减到 1。

示例 2：

输入：nums = [9,6,1,6,2]

输出：4

提示：

1 <= nums.length <= 1000
1 <= nums[i] <= 1000

解题思路

首先要注意本题只允许递减数字，只需要考虑分别考虑偶数索引满足情况和奇数索引满足情况的情况，模拟即可。

代码

class Solution {

  public:

    int movesToMakeZigzag(vector<int> &nums) {

        if (nums.size() == 1)

            return 0;

        // 考虑偶数索引大的情况

        int cnt2 = 0;

        for (int i = 1; i < nums.size(); i += 2) {

            if (i + 1 < nums.size() && i - 1 >= 0) {

                int tmp = std::min(nums[i + 1], nums[i - 1]);

                if (tmp <= nums[i]) {

                    cnt2 += nums[i] - tmp + 1;

                }

            } else if (i + 1 >= nums.size()) {

                if (nums[i - 1] <= nums[i]) {

                    cnt2 += nums[i] - nums[i - 1] + 1;

                }

            } else {

                if (nums[i + 1] <= nums[i]) {

                    cnt2 += nums[i] - nums[i + 1] + 1;

                }

            }

        }

        // 考虑奇数索引大的情况

        int cnt4 = 0;

        for (int i = 0; i < nums.size(); i += 2) {

            if (i + 1 < nums.size() && i - 1 >= 0) {

                int tmp = std::min(nums[i + 1], nums[i - 1]);

                if (tmp <= nums[i]) {

                    cnt4 += nums[i] - tmp + 1;

                }

            } else if (i + 1 >= nums.size()) {

                if (nums[i - 1] <= nums[i]) {

                    cnt4 += nums[i] - nums[i - 1] + 1;

                }

            } else {

                if (nums[i + 1] <= nums[i]) {

                    cnt4 += nums[i] - nums[i + 1] + 1;

                }

            }

        }

        return std::min(cnt2, cnt4);

    }

};

巴特西

1144. 递减元素使数组呈锯齿状 (Medium)

问题描述

解题思路

代码

最新文章

热门文章