 CHEFARRP - Editorial

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

Simple

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:

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 ?

1 Like

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

@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.

5 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,2]