乘积最大子数组(一) 连续
问题描述
给你一个整数数组 nums ,请你找出数组中乘积最大的连续子数组(该子数组中至少包含一个数字),并返回该子数组所对应的乘积。
class Solution {
public:
int maxProduct(vector<int>& nums) {
int n = nums.size();
vector<vector<int>>dp(n,vector<int>(2));
//因为可能存在两种情况: 正负数 ,所以维护两个数组或者
//dp[][0] max dp[][1] min
dp[0][0] = nums[0];
dp[0][1] = nums[0];
//max(dp[i-1][a] * nums[i], nums[i]) 可以模拟存储 前n个最大的,
for (int i =1; i<n; ++i)
{
dp[i][0] = max(max(dp[i-1][0] * nums[i], dp[i-1][1] * nums[i]), nums[i]);
dp[i][1] = min(min(dp[i-1][0] * nums[i], dp[i-1][1] * nums[i]), nums[i]);
}
int ret = dp[0][0];
for (int i=1; i <n; ++i)
{
if(dp[i][0] > ret)
ret = dp[i][0];
}
return ret;
}
};
(二) 不连续且有跨度限制
问题描述
有 n 个学生站成一排,每个学生有一个能力值,牛牛想从这 n 个学生中按照顺序选取 k 名学生,要求相邻两个学生的位置编号的差不超过 d,使得这 k 个学生的能力值的乘积最大,你能返回最大的乘积吗?
输入描述
每个输入包含 1 个测试用例。每个测试数据的第一行包含一个整数 n (1 <= n <= 50),表示学生的个数,接下来的一行,包含 n 个整数,按顺序表示每个学生的能力值 ai(-50 <= ai <= 50)。接下来的一行包含两个整数,k 和 d (1 <= k <= 10, 1 <= d <= 50)。
分析:
因为有正有负,负负得正,所以要维护两个dp数组,一个存储最大,一个存储最小。
定义fm[k][i]表示当选中了k个学生,并且以第i个学生为结尾,所产生的最大乘积;
fn[k][i]表示 当选中了k个学生,并且以第i个学生为结尾,所产生的最小乘积;
那么fm[k+1][i+1]=max(fm[k][i]*stu[i+1],fn[k][i]*stu[i+1]),
即当选中了k个学生后,再选择第i+1编号学生,所产生的最大乘积;
然而,并不能保证上一次选择的就是第i个学生,所以要遍历子数组fm[k],
令j从i到1,并且j与i+1之间小于间隔D,遍历fm[k][j],以及fn[k][j];
同理fn[k+1][i+1]=min(fn[k][i]*stu[i+1],fm[k][i]*stu[i+1])。
最后,遍历一遍fm[K][i]求得最大值(i从1~N)。
注 : 1LL<<63位为0 一个longlong 类型的数
#include <stdio.h>
inline long long max(long long a,long long b){return (a>b?a:b);}
inline long long min(long long a,long long b){return (a>b?b:a);}
int main(){
int N,K,D,i,j,k;
long long stu[51],fm[11][51],fn[11][51],ans;
while(~scanf("%d",&N)){
for(i=0;i<N;scanf("%lld",&stu[++i]));
scanf("%d %d",&K,&D);
for(i=0;i<=K;++i)
for(j=0;j<=N;fm[i][j]=fn[i][j]=0,++j);
for(i=1,ans=~1LL<<63;i<=N;++i){
fm[1][i]=fn[1][i]=stu[i];
for(k=2;k<=K;++k){
for(j=i-1;j>0 && i-j<=D;--j){
fm[k][i]=max(fm[k][i],max(fm[k-1][j]*stu[i],fn[k-1][j]*stu[i]));
fn[k][i]=min(fn[k][i],min(fn[k-1][j]*stu[i],fm[k-1][j]*stu[i]));
}
}
ans=max(ans,fm[K][i]);
}
printf("%lld\n",ans);
}
return 0;
}