小q和他的老婆小z最近开了一家花店,他们准备把店里最好看的花都摆在橱窗里。
但是他们有很多花瓶,每个花瓶都具有各自的特点,因此,当各个花瓶中放入不同的花束时,会产生不同的美学效果。
为了使橱窗里的花摆放的最合适,他们得想个办法安排每种花的摆放位置。
可是因为小q和小z每天都太忙,没有时间设计橱窗里花的摆法,所以他们想让你帮他们求出花摆放的最大美观程度和每种花所放的位置。
每种花都有一个标识,假设杜鹃花的标识数为1,秋海棠的标识数为2,康乃馨的标识数为3,所有的花束在放入花瓶时必须保持其标识数的顺序,即:
杜鹃花必须放在秋海棠左边的花瓶中,秋海棠必须放在康乃馨左边的花瓶中。
如果花瓶的数目大于花束的数目。则多余的花瓶必须空置,且每个花瓶中只能放一束花。
每种花放在不同的瓶子里会产生不同的美观程度,美观程度可能是正数也可能是负数。
上述例子中,花瓶与花束的不同搭配所具有的美观程度,如下表所示:
花 瓶
1 2 3 4 5
1 (杜鹃花) 7 23 -5 -24 16
2 (秋海棠) 5 21 -4 10 23
3 (康乃馨) -21 5 -4 -20 20
根据上表,杜鹃花放在花瓶2中,会显得非常好看;但若放在花瓶4中则显得十分难看。
为取得最大美观程度,你必须在保持花束顺序的前提下,使花束的摆放取得最大的美学值,并求出每种花应该摆放的花瓶的编号。
输入格式
第1行:两个整数F和V,表示共有F种花,V个花瓶。
第2行到第F+1行:每行有V个数,表示花摆放在不同花瓶里的美观程度值value。(美观程度和小于2^31,美观程度有正有负)
输出格式
输出有两行:第一行为输出最大美观程度和的值,第二行有F个数表示每朵花应该摆放的花瓶的编号。
若有多种方案,输出字典序较小的方案(美观程度不变的情况下,花尽量往前放)。
数据范围
1≤F≤V≤100,
输入样例:
3 5
7 23 -5 -24 16
5 21 -4 10 23
-21 5 -4 -20 20
输出样例:
53
2 4 5
思路:(线性dp+暴力枚举+还原方案)
首先,这显然是一个线性dp问题,接下来就要考虑状态表示和状态计算:
状态表示:
用f[i][j]表示放置前i朵花,且当前第i朵放在j号花瓶时的最大价值
状态计算:
按当前第i朵花以及前一朵(第i-1朵)放的位置的不同可以将决策方案的集合划分为若干项,且第i朵应该由第i-1朵放的位置转移过来,考虑到题目要求花必须按顺序摆放,所以即可确定出第i朵以及第i-1朵的枚举范围:
因为要按花的编号顺序放置,所以:对于第i朵,下限:理想情况:第1朵放在1,第二朵放在2,...这样第i朵最近要放在i;
上限:第i多最远放在m-(n-i) (因为要为后面n-i朵留出空);
同样的,第i-1朵的范围也就确定了,为:[i-1,j-1]
所以,当前f[i][j]的值即为:i-1朵花放在k且第i朵放在j时的最大值 k:[i-1,j-1], j:[i,m-(n-i)]
即:
答案即为:
还原具体方案:
因为已经通过dp求出了答案,所以根据答案倒推回去,即把原来的dp过程反过来进行判断得出每次的具体决策,倒序存在答案集合中,然后顺序输出答案集合即可。
完整代码:
#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
const int maxn=110,inf=0x3f3f3f3f;
int f[maxn][maxn],val[maxn][maxn],ans[maxn];
int main()
{
int n,m;
cin>>n>>m;
for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++){
cin>>val[i][j];
f[i][j]=-inf;
}
}
for(int i=1;i<=n;i++){//枚举前i朵花
for(int j=i;j<=m-(n-i);j++){//枚举当前第i朵花能放的位置(因为要按花的编号顺序放置,所以:下限:理想情况:第1朵放在1,第二朵放在2,...这样第i朵最近要放在i;上限:第i朵最远放在m-(n-i)(因为要为后面n-i朵留出空))
for(int k=i-1;k<j;k++){//枚举前一朵花(i-1)能放的位置
f[i][j]=max(f[i][j],f[i-1][k]+val[i][j]);
}
}
}
int res=-inf;
for(int i=1;i<=m;i++) res=max(res,f[n][i]);
cout<<res<<endl;
int cnt=n;
for(int i=n;i;i--){
for(int j=1;j<=m;j++){//因为答案取字典序小的,所以位置从小到大枚举来找满足条件的即可
if(f[i][j]==res){
ans[cnt--]=j;
res-=val[i][j];
break;
}
}
}
for(int i=1;i<=n;i++) cout<<ans[i]<<" ";
cout<<endl;
return 0;
}