題目描述
給你一個(gè)整數(shù)數(shù)組 nums,每次 操作 會(huì)從中選擇一個(gè)元素并 將該元素的值減少 1。
如果符合下列情況之一,則數(shù)組 A 就是 鋸齒數(shù)組:
每個(gè)偶數(shù)索引對(duì)應(yīng)的元素都大于相鄰的元素,即 A[0] > A[1] < A[2] > A[3] < A[4] > ...
或者,每個(gè)奇數(shù)索引對(duì)應(yīng)的元素都大于相鄰的元素,即 A[0] < A[1] > A[2] < A[3] > A[4] < ...
返回將數(shù)組 nums 轉(zhuǎn)換為鋸齒數(shù)組所需的最小操作次數(shù)。
示例 1:
輸入:nums = [1,2,3]
輸出:2
解釋:我們可以把 2 遞減到 0,或把 3 遞減到 1。
示例 2:
輸入:nums = [9,6,1,6,2]
輸出:4
題目分析
- 每次操作只是對(duì)一個(gè)元素減1
- 操作時(shí)取前后的最大差值并加1
- 分別對(duì)偶數(shù)位和奇數(shù)位進(jìn)行操作并求和,取其最小值。
int movesToMakeZigzag(vector<int>& nums) {
int ans[2];
ans[1] = ans[0] = 0;
for(int i = 0; i < nums.size(); i++) {
int d1 =0,d2=0;
//對(duì)前一個(gè)元素判別需要減去多少
if(i > 0 && nums[i] >= nums[i-1] ) {
d1 = nums[i] - nums[i-1] + 1;
}
//對(duì)后一個(gè)元素判別需要減去多少
if(i < nums.size()-1 && nums[i] >= nums[i+1]) {
d2 = nums[i] - nums[i+1] + 1;
}
//取前后最大值并求和。
ans[i%2] += max(d1,d2);
}
return min(ans[1], ans[0]);
}