题目背景
SOURCE:NOIP2015-SHY-9
题目描述
给定 2 个正整数序列
A1
,
A2
,序列长度分别为
L1
,
L2
。
你可以进行以下的一次操作:
1. 选择两个数
K1
,
K2
(1≤
K1
≤
L1
, 1≤
K2
≤
L2
);
2. 移去
A1
中最后
K1
个数,得到这
K1
个数的和
S1
,
L1
对应减少
K1
;
3. 移去
A2
中最后
K2
个数,得到这
K2
个数的和
S2
,
L2
对应减少
K2
;
此次操作的费用为:(
S1
-
K1
) * (
S2
-
K2
)。
进行以上操作直至两个序列都为空,求最小的费用总和。
注意:序列为空当且仅当两个序列同时为空。
输入格式
第一行是两个正整数
L1
和
L2
,表示
A1
与
A2
的长度。
第二行
L1
个整数,表示序列
A1[1..L1]
。
第三行
L2
个整数,表示序列
A2[1..L2]
。
输出格式
输出一个整数,表示最小费用。
样例数据
输入
3 2
1 2 3
1 2
输出
2
备注
【样例说明】
第一次选取
K1
=1,
K2
=1。费用为 (3-1)*(2-1) = 2。
第二次选取
K1
=2,
K2
=1。费用为 (1+2-2)*(1-1) = 0。
所以,总费用为 2。
【数据范围】
对 20% 的输入数据:1≤
L1
,
L2
≤20
对 40% 的输入数据:1≤
L1
≤400;1≤
L2
≤150
对 100% 的输入数据:1≤
L1
,
L2
,
A1[1..L1]
,
A2[1..L2]
≤5,000
分析:令f[i][j]表示S1还剩下i个数字、S2还剩下j个数字的最小花费。
那么转移方程为f[i-k1][j-k2]=f[i][j]+(sum1-k1)*(sum2-k2)。f[0][0]即为答案。
注意到后面添加项是有办法去除掉K1和K2的。通过将两个序列中所有数-1可以优化方程为f[i-k1][j-k2]=f[i][j]+sum1*sum2
现在讨论sum1*sum2,通过乘法分配律展开sum1*sum2,为:
a1[i]*a2[j]+a1[i]*a2[j-1]+…+a1[i]*a2[j-k2+1]+a1[i-1]*a2[j]+…+a1[i-k1+1]*a2[i-k2+1]。
可转化方程为最后标准结果转移方程:
f[i][j]=min(f[i+1][j+1],f[i+1][j],f[i][j+1])+a1[i+1]*a2[j+1]
(由于(a+b)*(c+d)>ac+bd,可以确定取数时有一边肯定只取一个,不然不是最优)
代码
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<ctime>
#include<cmath>
#include<algorithm>
#include<cctype>
#include<iomanip>
#include<queue>
#include<set>
using namespace std;
int getint()
{
int sum=0,f=1;
char ch;
for(ch=getchar();!isdigit(ch)&&ch!='-';ch=getchar());
if(ch=='-')
{
f=-1;
ch=getchar();
}
for(;isdigit(ch);ch=getchar())
sum=(sum<<3)+(sum<<1)+ch-48;
return sum*f;
}
int f[5050][5050];
int l1,l2;
int a[5050];
int b[5050];
int main()
{
freopen("array.in","r",stdin);
freopen("array.out","w",stdout);
cin>>l1>>l2;
for(int i=1;i<=l1;i++)
{
cin>>a[i];
a[i]--;
}
for(int i=1;i<=l2;i++)
{
cin>>b[i];
b[i]--;
}
memset(f,127,sizeof(f));
f[l1][l2]=0;
for(int i=l1-1;i>=0;i--)
{
for(int j=l2-1;j>=0;j--)
{
f[i][j]=min(f[i+1][j],min(f[i+1][j+1],f[i][j+1]))+a[i+1]*b[j+1];
}
}
cout<<f[0][0];
}
本题结。