代码引自leetcode题解库,我算是一个新手,先看懂代码为主。
代码作者:slowbird
class Solution {
public int movesToMakeZigzag(int[] nums) {
if (null == nums || nums.length == 0) return -1;
if (nums.length == 1) return 0;
// gap 表示当前元素与相邻元素的差值
// 差值含义: 为正表示需要减少的操作次数
// 为负则不用理会(已经符合条件)
// evenSum 表示偶数索引符合条件需要的操作次数之和
// oddSum 表示奇数索引符合条件需要的操作次数之和
// n 表示数组长度
int gap = 0, evenSum = 0, oddSum = 0, n = nums.length;
// 计算偶数索引需要减少多少次数才能符合条件
for (int i = 0; i < n; i = i+2) {
gap = 0;
if (i != 0) gap = Math.max(gap, nums[i] - nums[i-1] + 1);
if (i+1 < n) gap = Math.max(gap, nums[i] - nums[i+1] + 1);
evenSum += gap;
}
// 计算奇数索引需要减少多少次数才能符合条件
for (int i = 1; i < n; i = i+2) {
gap = 0;
if (i != 0) gap = Math.max(gap, nums[i] - nums[i-1] + 1);
if (i+1 < n) gap = Math.max(gap, nums[i] - nums[i+1] + 1);
oddSum += gap;
}
return Math.min(evenSum, oddSum);
}
}
作者:lingsh
链接:https://leetcode-cn.com/problems/decrease-elements-to-make-array-zigzag/solution/tong-guo-dui-ou-shu-wei-qi-shu-wei-jin-xing-cao-zu/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
该算法十分精妙,通过gap记录奇/偶数的递增/递减次数。
例如
if (i != 0) gap = Math.max(gap, nums[i] - nums[i-1] + 1);
if (i+1 < n) gap = Math.max(gap, nums[i] - nums[i+1] + 1);
这里的gap初始化为0,举个例子a[1],a[2],a[3],用a[2]-a[1]和a[2]-a[3],通过max函数取最大的变化次数,至于为什么后面要加一,是因为3-2等于1对吧,但是3要比2小,那么还要递减一次,所以要加一个1.