leetcode题目4
题目描述
给你一个整数数组 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
解答
1.用一个内联函数
2.利用按位&来判断奇偶
3.用fllag判断是奇数位还是偶数位
4.最终答案就是取奇数位和偶数位两种情况的最小值
class Solution {
public:
inline int solve(vector<int>& nums,int flag){
int n=nums.size();
int ans=0;
vector<int> now=nums;
for(int i=0;i<n;i++){
if((i&1)==flag){
while(i>0&&now[i-1]>=now[i]){
++ans;
--now[i-1];
}
while(i+1<n&&now[i+1]>=now[i]){
++ans;
--now[i+1];
}
}
}
return ans;
}
int movesToMakeZigzag(vector<int>& nums) {
return min(solve(nums,0),solve(nums,1));
}
};