← All problems

3355. Zero Array Transformation I

MediumOpen on LeetCodeProblem statement

Problem Statement

3355. Zero Array Transformation I

Medium


You are given an integer array nums of length n and a 2D array queries, where queries[i] = [li, ri].

For each queries[i]:

A Zero Array is an array where all elements are equal to 0.

Return true if it is possible to transform nums into a Zero Array after processing all the queries sequentially, otherwise return false.

 

Example 1:

Input: nums = [1,0,1], queries = [[0,2]]

Output: true

Explanation:

Example 2:

Input: nums = [4,3,2,1], queries = [[1,3],[0,2]]

Output: false

Explanation:

 

Constraints:

Java

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