【从零开始的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

我的解答

java 复制代码
class Solution {
    public int countValidSelections(int[] nums) {
        int sum = 0;
        for(int num : nums){
            sum += num;
        }
        /**
         * 由示例可知,满足条件的curr左右两边差值必须小于等于1
         * 等于0,即curr左右两边的值相等时,向左向右就可
         * 等于1,即curr左右两边的值相差为1时,仅能先向值大的方向开始移动
         */
        int res = 0,pre = 0;
        for(int num : nums){
            if(num > 0){
                pre += num;
                if(pre * 2 > sum + 1) break;
            }else if(pre * 2 == sum){
                res += 2;
            }else if(Math.abs(pre * 2 - sum) == 1){
                res++;
            }
        }
        return res;
    }
}
相关推荐
THMAIL4 分钟前
深度学习从入门到精通 - LSTM与GRU深度剖析:破解长序列记忆遗忘困境
人工智能·python·深度学习·算法·机器学习·逻辑回归·lstm
JuneXcy8 分钟前
结构体简介
c语言·数据结构·算法
ST.J11 分钟前
swing笔记
java·笔记
jiaway13 分钟前
【C语言】第四课 指针与内存管理
c语言·开发语言·算法
菩提树下的凡夫17 分钟前
瑞芯微RV1126目标识别算法Yolov8的部署应用
java·算法·yolo
爱隐身的官人26 分钟前
新后端漏洞(上)- Java RMI Registry反序列化漏洞
java·反序列化漏洞
叫我阿柒啊30 分钟前
从Java全栈到前端框架:一次真实的面试对话与技术解析
java·javascript·typescript·vue·springboot·react·前端开发
晚安里40 分钟前
Spring 框架(IoC、AOP、Spring Boot) 的必会知识点汇总
java·spring boot·spring
爱隐身的官人1 小时前
新后端漏洞(上)- Aapache Tomcat AJP 文件包含漏洞(CVE-2020-1938)
java·tomcat·ajp
多打代码1 小时前
2025.09.05 用队列实现栈 & 有效的括号 & 删除字符串中的所有相邻重复项
python·算法