[贪心] leetcode 1144 Decrease Elements To Make Array Zigzag

最新推荐文章于 2024-09-18 23:25:39 发布

ayzejx8547408

最新推荐文章于 2024-09-18 23:25:39 发布

阅读量96

点赞数

文章标签：数据结构与算法

原文链接：http://www.cnblogs.com/fish1996/p/11297968.html

版权

problem:https://leetcode.com/contest/weekly-contest-148/problems/decrease-elements-to-make-array-zigzag/

贪心题。有两种可能的zigzag，一个是奇数下标为最大值，一个是偶数下标为最大值。分别就这两种情况计算可能的最小decrease次数。

对于a[i] >= a[i-1]，如果希望a[i] < a[i-1], 使得a[i] = a[i-1] - 1即可。

以下为比赛期间的代码，可能有可以精简的地方：

（这道题一开始不是easy么，怎么突然变成medium了)

class Solution {
public:
    int movesToMakeZigzag(vector<int>& nums) {
        
        int oddMax = 0;
        vector<int> oddNums(nums);
        for(int i = 1;i < oddNums.size();i++)
        {
            if(i % 2)
            {
                if(oddNums[i] <= oddNums[i - 1])
                {
                    oddMax += (oddNums[i - 1] - oddNums[i]) + 1;
                    oddNums[i - 1] = oddNums[i] - 1;
                }
            }
            else 
            {
                if(oddNums[i] >= oddNums[i - 1])
                {
                    oddMax += (oddNums[i] - oddNums[i - 1]) + 1;
                    oddNums[i] = oddNums[i - 1] - 1;
                }               
            }
        }
        int evenMax = 0;
        vector<int> evenNums(nums);
        for(int i = 1;i < evenNums.size();i++)
        {
            if(i % 2)
            {
                if(evenNums[i] >= evenNums[i - 1])
                {
                    evenMax += (evenNums[i] - evenNums[i - 1]) + 1;
                    evenNums[i] = evenNums[i - 1] - 1;
                }
            }
            else 
            {
                if(evenNums[i] <= evenNums[i - 1])
                {
                    evenMax += (evenNums[i - 1] - evenNums[i]) + 1;
                    evenNums[i - 1] = evenNums[i] - 1;
                }               
            }           
        }
   //     cout << oddMax << " " << evenMax << endl;
        return min(oddMax, evenMax);
    }
};