CHEFARRP - Editorial

PROBLEM LINK:

Practice
Contest

Author: Misha Chorniy
Tester: Jingbo Shang
Editorialist: Pushkar Mishra

DIFFICULTY:

Simple

PREREQUISITES:

None

PROBLEM:

Given an array, we have to report the number of subarrays such that the product of all numbers in that subarray is equal to the sum of all numbers in that subarray.

EXPLANATION:

The problem is very direct given the size of the input array. Since n \leq 50, we can directly iterate over the subarrays and calculate product and sum. The editorialist’s solution is very clear in describing the approach. Below is the pseudo-code of the same:

function get_ans(input_array[n])
{
    int count = 0;
    for(i = 1 to n)
    {
        int sum = 0, product = 1;

        for(j = i down_to 1)
        {
            //considering all subarrays ending at i.

            sum = sum + input_array[j];
            product = product*input_array[j];

            if(sum == product)
                count = count+1;
        }
    }

    return count;
}

COMPLEXITY:

\mathcal{O}(n^2) per test case.

SAMPLE SOLUTIONS:

Author
Tester
Editorialist

1 Like

I did the same way and got correct answer. But what if n would have been a large number? What is a faster(in terms of time complexity) way to do it ?

10 Likes

@infinitparadox

Good question, I think in this problem nothing faster than O(n^2). If someone has something better, I will glad to hear it.

1 Like

@mgch

Actually.

Since there can’t be too many numbers greater than 1 in any such subarray (since the product grows exponentially, O(\log{sum}) at most), you can store just the numbers greater than 1 in another array and try all their subarrays - there are O(N\log{sum}) of them. For each such subarray, you know the product, which the ignored 1-s don’t change; this subarray corresponds to subarrays [a,b] in the original array, where a and b must be in some, disjoint, ranges (only up to the first integer > 1 to the left/right); the sum is a linear function of b-a and since it has to be equal to the product, any a for which we have a valid b gives one of the counted subarrays; we just need to find the range of such a-s.

For a given subarray of integers > 1, everything can be implemented in O(1), so we have O(N\log{sum}) time complexity.

6 Likes

How to solve the same problem for larger numbers and strict time limit!!!

I have submitted the sol.
but m getting a nzec.
I am using string.split("\s+").
Any suggestions ?

Pardon me if i am missing something elementary, but wouldn’t this fail with an inputs such as [2,3,2]?

This algo will not be able to generate 2,2 as a possible option and will return the result as 3. The result should be 4. [2], [3],[2],[2,2]

5 Likes

Pasting a reply from Stackoverflow about what is a subarray

“I think the subarray means the array in which the contiguousness is based on the index . for example lets take an example
2 3 1 3 1 (index are 0,1,2,3,4 respectively )
so subarray is 2,3,1 3 (index 0,1,2,3) But (1,2 1) (index 2,0,4) cannot be a subarray coz the indexes are non-contiguous …”

Your example [2, 3 ,2]
The [2, 2] array has indexes 0, 2 which are non-contiguous. Therefore, it can’t be an option for subarray

5 Likes

this editorial isn’t completely correct,what if some random number position elements yield such type of sub-array for ex like a[o],a[3],a[9].this case i not incuded in the editorial

@ajaman13 see the explanation by @povi

nice explanation…

It will fail because [2,2] is not a subarray of [2,3,2].The subarray of [2,3,2] includes = {[2],[3],[2],[2,3],[3,2],[2,3,2]}.The problem is asking for number of subarrays.Indeed [2,2] is a subsequence of [2,3,2]

1 Like

Continuing the discussion from CHEFARRP - Editorial:
@mgch
if array is 1 2 3 4 5
sub arrays starting with 1 will be 1 12 123 12345
once any subarray of a index has products value greater than sum we no longer need to check the remaining sub arrays as it goes on increasing and they will never be equal
so this may decrease the complexity I think

use try catch

Example case 1. There are 4 such subarrays: A [1…1], A [2…2], A [3…3], A [1…3]. Consider A [1…3], sum = 1 + 3 + 2 = 6, product = 1 * 3 * 2 = 6.

Can someone explain how A [3…3] has come here. It will be really appreciated.
Thanks

I generated all subaarays for the last test cases and they are [(1,), (2,), (2,), (2,), (2,), (1,), (2, 2), (2, 2), (2, 2), (2, 2), (2, 2), (2, 2), (1, 2, 2, 2, 1), (1, 2, 2, 2, 1), (1, 2, 2, 2, 1), (1, 2, 2, 2, 1)] . On a manual inspection i can find the number is 16 not 9.

What am I missing here?

It means contiguous subarrays. I do agree the question could have been worded more clearly, but the examples were designed really well to demonstrate that is what was intended.