k 尺寸子阵列的最大乘积
原文:https://www . geesforgeks . org/maximum-product-subarray-size-k/
给定一个由 n 个正整数和一个整数 k 组成的数组,求大小为 k 的最大乘积子数组,即求数组中 k 个连续元素的最大乘积,其中 k <= n. 例:
Input: arr[] = {1, 5, 9, 8, 2, 4,
1, 8, 1, 2}
k = 6
Output: 4608
The subarray is {9, 8, 2, 4, 1, 8}
Input: arr[] = {1, 5, 9, 8, 2, 4, 1, 8, 1, 2}
k = 4
Output: 720
The subarray is {5, 9, 8, 2}
Input: arr[] = {2, 5, 8, 1, 1, 3};
k = 3
Output: 80
The subarray is {2, 5, 8}
方法 1(简单:O(n*k)) 一种简单的方法是逐个考虑 k 大小的所有子阵列。这种方法需要两个循环,因此复杂度为 0(n * k)。 方法 2(高效:O(n)) 我们可以在 O(n)中求解,利用这样一个事实:如果我们有以前子阵列的乘积可用,那么大小为 k 的子阵列的乘积可以在 O(1)时间内计算出来。
curr_product = (prev_product / arr[i-1]) * arr[i + k -1]
prev_product : Product of subarray of size k beginning
with arr[i-1]
curr_product : Product of subarray of size k beginning
with arr[i]
这样,我们可以只在一次遍历中计算出最大 k 大小的子阵积。下面是 C++实现的思路。
C++
// C++ program to find the maximum product of a subarray
// of size k.
#include <bits/stdc++.h>
using namespace std;
// This function returns maximum product of a subarray
// of size k in given arrar, arr[0..n-1]. This function
// assumes that k is smaller than or equal to n.
int findMaxProduct(int arr[], int n, int k)
{
// Initialize the MaxProduct to 1, as all elements
// in the array are positive
int MaxProduct = 1;
for (int i=0; i<k; i++)
MaxProduct *= arr[i];
int prev_product = MaxProduct;
// Consider every product beginning with arr[i]
// where i varies from 1 to n-k-1
for (int i=1; i<=n-k; i++)
{
int curr_product = (prev_product/arr[i-1]) *
arr[i+k-1];
MaxProduct = max(MaxProduct, curr_product);
prev_product = curr_product;
}
// Return the maximum product found
return MaxProduct;
}
// Driver code
int main()
{
int arr1[] = {1, 5, 9, 8, 2, 4, 1, 8, 1, 2};
int k = 6;
int n = sizeof(arr1)/sizeof(arr1[0]);
cout << findMaxProduct(arr1, n, k) << endl;
k = 4;
cout << findMaxProduct(arr1, n, k) << endl;
int arr2[] = {2, 5, 8, 1, 1, 3};
k = 3;
n = sizeof(arr2)/sizeof(arr2[0]);
cout << findMaxProduct(arr2, n, k);
return 0;
}
Java 语言(一种计算机语言,尤用于创建网站)
// Java program to find the maximum product of a subarray
// of size k
import java.io.*;
import java.util.*;
class GFG
{
// Function returns maximum product of a subarray
// of size k in given arrar, arr[0..n-1]. This function
// assumes that k is smaller than or equal to n.
static int findMaxProduct(int arr[], int n, int k)
{
// Initialize the MaxProduct to 1, as all elements
// in the array are positive
int MaxProduct = 1;
for (int i=0; i<k; i++)
MaxProduct *= arr[i];
int prev_product = MaxProduct;
// Consider every product beginning with arr[i]
// where i varies from 1 to n-k-1
for (int i=1; i<=n-k; i++)
{
int curr_product = (prev_product/arr[i-1]) *
arr[i+k-1];
MaxProduct = Math.max(MaxProduct, curr_product);
prev_product = curr_product;
}
// Return the maximum product found
return MaxProduct;
}
// driver program
public static void main (String[] args)
{
int arr1[] = {1, 5, 9, 8, 2, 4, 1, 8, 1, 2};
int k = 6;
int n = arr1.length;
System.out.println(findMaxProduct(arr1, n, k));
k = 4;
System.out.println(findMaxProduct(arr1, n, k));
int arr2[] = {2, 5, 8, 1, 1, 3};
k = 3;
n = arr2.length;
System.out.println(findMaxProduct(arr2, n, k));
}
}
// This code is contributed by Pramod Kumar
Python 3
# Python 3 program to find the maximum
# product of a subarray of size k.
# This function returns maximum product
# of a subarray of size k in given arrar,
# arr[0..n-1]. This function assumes
# that k is smaller than or equal to n.
def findMaxProduct(arr, n, k) :
# Initialize the MaxProduct to 1,
# as all elements in the array
# are positive
MaxProduct = 1
for i in range(0, k) :
MaxProduct = MaxProduct * arr[i]
prev_product = MaxProduct
# Consider every product beginning
# with arr[i] where i varies from
# 1 to n-k-1
for i in range(1, n - k + 1) :
curr_product = (prev_product // arr[i-1]) * arr[i+k-1]
MaxProduct = max(MaxProduct, curr_product)
prev_product = curr_product
# Return the maximum product found
return MaxProduct
# Driver code
arr1 = [1, 5, 9, 8, 2, 4, 1, 8, 1, 2]
k = 6
n = len(arr1)
print (findMaxProduct(arr1, n, k) )
k = 4
print (findMaxProduct(arr1, n, k))
arr2 = [2, 5, 8, 1, 1, 3]
k = 3
n = len(arr2)
print(findMaxProduct(arr2, n, k))
# This code is contributed by Nikita Tiwari.
C
// C# program to find the maximum
// product of a subarray of size k
using System;
class GFG
{
// Function returns maximum
// product of a subarray of
// size k in given arrar,
// arr[0..n-1]. This function
// assumes that k is smaller
// than or equal to n.
static int findMaxProduct(int []arr,
int n, int k)
{
// Initialize the MaxProduct
// to 1, as all elements
// in the array are positive
int MaxProduct = 1;
for (int i = 0; i < k; i++)
MaxProduct *= arr[i];
int prev_product = MaxProduct;
// Consider every product beginning
// with arr[i] where i varies from
// 1 to n-k-1
for (int i = 1; i <= n - k; i++)
{
int curr_product = (prev_product /
arr[i - 1]) *
arr[i + k - 1];
MaxProduct = Math.Max(MaxProduct,
curr_product);
prev_product = curr_product;
}
// Return the maximum
// product found
return MaxProduct;
}
// Driver Code
public static void Main ()
{
int []arr1 = {1, 5, 9, 8, 2,
4, 1, 8, 1, 2};
int k = 6;
int n = arr1.Length;
Console.WriteLine(findMaxProduct(arr1, n, k));
k = 4;
Console.WriteLine(findMaxProduct(arr1, n, k));
int []arr2 = {2, 5, 8, 1, 1, 3};
k = 3;
n = arr2.Length;
Console.WriteLine(findMaxProduct(arr2, n, k));
}
}
// This code is contributed by anuj_67.
服务器端编程语言(Professional Hypertext Preprocessor 的缩写)
<?php
// PHP program to find the maximum
// product of a subarray of size k.
// This function returns maximum
// product of a subarray of size
// k in given arrar, arr[0..n-1].
// This function assumes that k
// is smaller than or equal to n.
function findMaxProduct( $arr, $n, $k)
{
// Initialize the MaxProduct to
// 1, as all elements
// in the array are positive
$MaxProduct = 1;
for($i = 0; $i < $k; $i++)
$MaxProduct *= $arr[$i];
$prev_product = $MaxProduct;
// Consider every product
// beginning with arr[i]
// where i varies from 1
// to n-k-1
for($i = 1; $i < $n - $k; $i++)
{
$curr_product = ($prev_product / $arr[$i - 1]) *
$arr[$i + $k - 1];
$MaxProduct = max($MaxProduct, $curr_product);
$prev_product = $curr_product;
}
// Return the maximum
// product found
return $MaxProduct;
}
// Driver code
$arr1 = array(1, 5, 9, 8, 2, 4, 1, 8, 1, 2);
$k = 6;
$n = count($arr1);
echo findMaxProduct($arr1, $n, $k),"\n" ;
$k = 4;
echo findMaxProduct($arr1, $n, $k),"\n";
$arr2 = array(2, 5, 8, 1, 1, 3);
$k = 3;
$n = count($arr2);
echo findMaxProduct($arr2, $n, $k);
// This code is contributed by anuj_67.
?>
java 描述语言
<script>
// JavaScript program to find the maximum
// product of a subarray of size k
// Function returns maximum
// product of a subarray of
// size k in given arrar,
// arr[0..n-1]. This function
// assumes that k is smaller
// than or equal to n.
function findMaxProduct(arr, n, k)
{
// Initialize the MaxProduct
// to 1, as all elements
// in the array are positive
let MaxProduct = 1;
for (let i = 0; i < k; i++)
MaxProduct *= arr[i];
let prev_product = MaxProduct;
// Consider every product beginning
// with arr[i] where i varies from
// 1 to n-k-1
for (let i = 1; i <= n - k; i++)
{
let curr_product =
(prev_product / arr[i - 1]) * arr[i + k - 1];
MaxProduct = Math.max(MaxProduct, curr_product);
prev_product = curr_product;
}
// Return the maximum
// product found
return MaxProduct;
}
let arr1 = [1, 5, 9, 8, 2, 4, 1, 8, 1, 2];
let k = 6;
let n = arr1.length;
document.write(findMaxProduct(arr1, n, k) + "</br>");
k = 4;
document.write(findMaxProduct(arr1, n, k) + "</br>");
let arr2 = [2, 5, 8, 1, 1, 3];
k = 3;
n = arr2.length;
document.write(findMaxProduct(arr2, n, k) + "</br>");
</script>
输出:
4608
720
80
本文由 阿舒托什·库马尔 供稿。如果你喜欢 GeeksforGeeks 并想投稿,你也可以使用write.geeksforgeeks.org写一篇文章或者把你的文章邮寄到 contribute@geeksforgeeks.org。看到你的文章出现在极客博客主页上,帮助其他极客。 如果发现有不正确的地方,或者想分享更多关于上述话题的信息,请写评论。
版权属于:月萌API www.moonapi.com,转载请注明出处