动态规划 luogu P1854 花店橱窗布置

题目大意

P1854
给你$n$束花和$m$个花瓶，第$i$束花放在第$j$个花瓶中可以取得$a_{i,j}$的好看值，并且第$i$束花必须放在第$i+1$束花左边。每个花瓶仅能放一束花，花瓶的数量可能大于花的数量，求出最大可以获得多少好看值以及第$i$朵花应该放到哪一个花瓶中。

题目描述

首先我们来处理最大好看值。当看到第$i$束花必须放在第$i+1$束花左边时，就应该想到dp的无后效性。定义dp数组$f$，$f_{i,j}$表示前$i$束花放在前$j$个花瓶时能获得的最大好看值，我们这时只需要考虑第$i$束花需不需要放在第$j$个花瓶里，若是放，则$f_{i,j}=f_{i-1,j-1}+a_{i,j}$，即前$i-1$束花放在前$j-1$个花瓶里得到的最大值加上当前第$i$束花放在第$j$个花瓶时的值。若是不放，则$f_{i,j}=f_{i,j-1}$。
再考虑放的花瓶的位置，位置其实在处理好看值时就可以顺便求出来，因为第$f_{i,j}$已经是当前$i$束花放在前$j$个花瓶里时可获得的最大好看值了，所以只需要记录得到$f_{i,j}$时放的花瓶的位置即可。因此定义一个结构体$w$，$w{.}_{i,j}$中的$a$数组即为前$i$束花放在前$j$个花瓶时每个放了花的花瓶的位置。

代码

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
using namespace std;
struct ap{
	int a[110],l;//l为放了花的花瓶的数量
}w[110][110];
int a[110][110],f[110][110],n,m;
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	cin>>n>>m;
	for(int i=1;i<=n;i++)
		for(int j=1;j<=m;j++)
			cin>>a[i][j];
	memset(f,-127,sizeof(f));
	for(int i=0;i<=m;i++)
		f[0][i]=0;
	for(int i=1;i<=n;i++)
		for(int j=i;j<=m;j++)
		{
			if(f[i-1][j-1]+a[i][j]>f[i][j-1])
			{
				w[i][j]=w[i-1][j-1];
				w[i][j].a[++w[i][j].l]=j;//多放置一个花瓶
				f[i][j]=f[i-1][j-1]+a[i][j];
			}else{
				w[i][j]=w[i][j-1];//不放
				f[i][j]=f[i][j-1];
			}
		}
	cout<<f[n][m]<<endl;
	for(int i=1;i<=w[n][m].l;i++)
		cout<<w[n][m].a[i]<<" ";//输出最后的方案
	return 0;
}