【Leetcode 每日一题】3355. 零数组变换 I

最新推荐文章于 2025-08-25 14:30:19 发布

原创最新推荐文章于 2025-08-25 14:30:19 发布 · 565 阅读

15 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #数据结构

Leetcode Daily 专栏收录该内容

195 篇文章

订阅专栏

问题背景

给定一个长度为 $n$ 的整数数组 $n u m s$ 和一个二维数组 $q u er i es$ ，其中 $queries[i] = [l_i, r_i]$ 。
对于每个查询 $q u er i es [i]$ ：

在 $n u m s$ 的下标范围 $l_i, r_i]$ 内选择一个下标子集。
将选中的每个下标对应的元素值减 $1$ 。

零数组 是指所有元素都等于 $0$ 的数组。
如果在按顺序处理所有查询后，可以将 $n u m s$ 转换为 零数组 ，则返回 $t r u e$ ，否则返回 $f a l se$ 。

数据约束

$\le nums.length \le 10 ^ 5$
$\le nums[i] \le 10 ^ 5$
$\le queries.length \le 10 ^ 5$
$q u er i es [i] . l e n g t h = 2$
$\le l_i \le r_i < nums.length$

解题过程

由于操作的过程中可以选择子集，所以实际上问题可以转化为能否对范围上的数进行操作，使得所有元素均非正。
而对某个区间上的所有元素进行某种同样的操作，可以用差分来实现。

具体实现

class Solution {
    public boolean isZeroArray(int[] nums, int[][] queries) {
        int n = nums.length;
        int[] diff = new int[n + 1];
        for (int[] query : queries) {
            diff[query[0]]++;
            diff[query[1] + 1]--;
        }
        int num = 0;
        for (int i = 0; i < n; i++) {
            num += diff[i];
            if (nums[i] > num) {
                return false;
            }
        }
        return true;
    }
}