leetcode 3354. 使数组元素等于零 简单

给你一个整数数组 nums 。

开始时，选择一个满足 nums[curr] == 0 的起始位置 curr ，并选择一个移动 方向 ：向左或者向右。

此后，你需要重复下面的过程：

• 如果 curr 超过范围 [0, n - 1] ，过程结束。
• 如果 nums[curr] == 0 ，沿当前方向继续移动：如果向右移，则 递增 curr ；如果向左移，则 递减 curr 。
• 如果 nums[curr] > 0:
  • 将 nums[curr] 减 1 。
  • 反转 移动方向（向左变向右，反之亦然）。
  • 沿新方向移动一步。

如果在结束整个过程后，nums 中的所有元素都变为 0 ，则认为选出的初始位置和移动方向 有效 。

返回可能的有效选择方案数目。

示例 1：

输入：nums = [1,0,2,0,3]

输出：2

解释：

可能的有效选择方案如下：

• 选择 curr = 3 并向左移动。
  • [1,0,2,0,3] -> [1,0,2,0,3] -> [1,0,1,0,3] -> [1,0,1,0,3] -> [1,0,1,0,2] -> [1,0,1,0,2] -> [1,0,0,0,2] -> [1,0,0,0,2] -> [1,0,0,0,1] -> [1,0,0,0,1] -> [1,0,0,0,1] -> [1,0,0,0,1] -> [0,0,0,0,1] -> [0,0,0,0,1] -> [0,0,0,0,1] -> [0,0,0,0,1] -> [0,0,0,0,0].
• 选择 curr = 3 并向右移动。
  • [1,0,2,0,3] -> [1,0,2,0,3] -> [1,0,2,0,2] -> [1,0,2,0,2] -> [1,0,1,0,2] -> [1,0,1,0,2] -> [1,0,1,0,1] -> [1,0,1,0,1] -> [1,0,0,0,1] -> [1,0,0,0,1] -> [1,0,0,0,0] -> [1,0,0,0,0] -> [1,0,0,0,0] -> [1,0,0,0,0] -> [0,0,0,0,0].

示例 2：

输入：nums = [2,3,4,0,4,1,0]

输出：0

解释：

不存在有效的选择方案。

提示：

• 1 <= nums.length <= 100
• 0 <= nums[i] <= 100
• 至少存在一个元素 i 满足 nums[i] == 0 。

分析：计算每一个位置的前缀和，后缀和。接着遍历数组 nums，如果 nums[i] == 0，判断它的前缀和与后缀和的差的绝对值是否小于等于 1，若小于等于，则说明当前位置满足题意，且绝对值等于 0 时，从该点想左向右出发均符合，答案 +2；若绝对值等于 1 则只有一个方向可以，答案 +1。遍历完，统计答案即可。

int countValidSelections(int* nums, int numsSize) {int pre[numsSize+5],last[numsSize+5];pre[0]=nums[0],last[numsSize-1]=nums[numsSize-1];for(int i=1;i<numsSize;++i){pre[i]=pre[i-1]+nums[i];last[numsSize-1-i]=last[numsSize-i]+nums[numsSize-1-i];}int len=numsSize,ans=0;for(int i=0;i<len;++i){if(nums[i]==0){int temp=abs(pre[i]-last[i]);if(temp<=1)ans+=2-temp;}}return ans;
}