Description
在短学期和算法导论课期间,我们一共讨论了四种求解最大子段和问题的方法,上一届的期末考试也考了交替出现的子段和问题,今天我们再拓展一下,研究一下最大的带权子段和问题,这个问题我们已经找到了很有意思的应用背景。例如已知三个权系数分别为1、-2、1,考虑从i开始的序列,带权的累加和是:a[i]-2a[i+1]+a[i+2]+a[i+3]-2a[i+4]+a[i+5]+…,你的任务是编程序计算最大的带权子段和。
Input
本问题有多组测试数据,对于每组测试数据,输入有三行,第一行是两个用空格隔开的n和m,其中1<=m<=100,m<=n<=100000;第二行是用空格隔开的m个整数,表示带权的系数;第三行是n个用空格隔开的整数,表示一个有n个元素的序列。输入保证中间运算过程不会超过int范围。
Output
对于每组测试数据,输出只有一行,即最大的子段和的值。
Sample Input
10 3
1 -2 1
10 -9 11 -2 -3 12 8 7 -10 6
Sample Output
63
代码
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=100010;
ll dp[maxn][110];
ll a[maxn],b[maxn];
int main()
{
int n,m;
while(~scanf("%d%d",&n,&m))
{
for(int i=1;i<=m;++i)scanf("%lld",&b[i]);
for(int i=1;i<=n;++i)scanf("%lld",&a[i]);
ll ans=-0x3f3f3f3f3f3f3f3f;
for(int i=1;i<=n;++i)
{
for(int j=1;j<=min(m,n);j++)
{
if(j==1)
{
if(i-1<m)
dp[i][j]=a[i]*b[j];
else
dp[i][j]=max(dp[i-1][m]+a[i]*b[j],a[i]*b[j]);
}
else
dp[i][j]=dp[i-1][j-1]+a[i]*b[j];
ans=max(ans,dp[i][j]);
}
}
printf("%lld\n",ans);
}
return 0;
}