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

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);
    }
};

猜你喜欢

转载自www.cnblogs.com/fish1996/p/11297968.html