📜  product 和 sum 相等的子数组的数量

📅  最后修改于: 2022-05-13 01:57:47.751000             🧑  作者: Mango

product 和 sum 相等的子数组的数量

给定一个包含 n 个数字的数组。我们需要计算乘积和元素总和相等的子数组的数量

例子:

Input  : arr[] = {1, 3, 2}
Output : 4
The subarrays are :
[0, 0] sum = 1, product = 1,
[1, 1] sum = 3, product = 3,
[2, 2] sum = 2, product = 2 and 
[0, 2] sum = 1+3+2=6, product = 1*3*2 = 6

Input : arr[] = {4, 1, 2, 1}
Output : 5

这个想法很简单,我们检查每个子数组的乘积和其元素的总和是否相等。如果是,则将计数器变量增加 1

C++
// C++ program to count subarrays with
// same sum and product.
#include
using namespace std;
 
// returns required number of subarrays
int numOfsubarrays(int arr[] , int n)
{
    int count = 0; // Initialize result
 
    // checking each subarray
    for (int i=0; i


Java
// Java program to count subarrays with
// same sum and product.
 
class GFG
{
    // returns required number of subarrays
    static int numOfsubarrays(int arr[] , int n)
    {
        int count = 0; // Initialize result
      
        // checking each subarray
        for (int i=0; i


Python3
# python program to
# count subarrays with
# same sum and product.
 
# returns required
# number of subarrays
def numOfsubarrays(arr,n):
 
    count = 0 # Initialize result
  
    # checking each subarray
    for i in range(n):
     
        product = arr[i]
        sum = arr[i]
        for j in range(i+1,n):
         
            # checking if product is equal
            # to sum or not
            if (product==sum):
                count+=1
  
            product *= arr[j]
            sum += arr[j]
         
  
        if (product==sum):
            count+=1
     
    return count
 
# Driver code
 
arr = [1,3,2]
n =len(arr)
print(numOfsubarrays(arr , n))
 
# This code is contributed
# by Anant Agarwal.


C#
// C# program to count subarrays
// with same sum and product.
using System;
class GFG {
     
    // returns required number
    // of subarrays
    static int numOfsubarrays(int []arr ,
                              int n)
    {
         
        // Initialize result
        int count = 0;
     
        // checking each subarray
        for (int i = 0; i < n; i++)
        {
            int product = arr[i];
            int sum = arr[i];
            for (int j = i + 1; j < n; j++)
            {
                 
                // checking if product is
                // equal to sum or not
                if (product == sum)
                    count++;
     
                product *= arr[j];
                sum += arr[j];
            }
     
            if (product == sum)
                count++;
        }
        return count;
    }
     
    // Driver Code
    public static void Main()
    {
        int []arr = {1,3,2};
        int n = arr.Length;
        Console.Write(numOfsubarrays(arr , n));
    }
}
 
// This code is contributed by Nitin Mittal.


PHP


Javascript


输出:

4

时间复杂度:O(n 2 )