现有f朵花,从1到f编号。还有v个花瓶,从1到v编号。将每个花均放入花瓶中,第i个花放入第j个花瓶中,会获得a[i][j]的收益,求总收益的最大值,及最大值时的方案。
动态规划,定义状态dp[i][j]为前i朵花放入前j个花瓶中的总收益的最大值,则dp[i][j]=max(dp[i-1][j-1]+a[i][j],dp[i][j-1])。
再记录数组from,from[i][j]表示dp[i][j]是从哪里转移过来的。
#include <cstdio>
#include <cstring>
int n,m;
int a[101][101];
int dp[101][101];
bool from[101][101];
int ans[100];
int main() {
int i,j;
memset(dp,0x80,sizeof(dp));
memset(from,0,sizeof(from));
scanf("%d%d",&n,&m);
for (i=1;i<=n;i++)
for (j=1;j<=m;j++)
scanf("%d",&a[i][j]);
for (j=0;j<=m;j++)
dp[0][j]=0;
for (i=1;i<=n;i++)
for (j=1;j<=m;j++) {
if (dp[i-1][j-1]+a[i][j]>dp[i][j-1]) {
dp[i][j]=dp[i-1][j-1]+a[i][j];
from[i][j]=true;
} else {
dp[i][j]=dp[i][j-1];
}
}
printf("%d\n",dp[n][m]);
for (i=n,j=m;j!=0;j--) {
if (from[i][j]) {
ans[i]=j;
i--;
}
}
for (i=1;i<=n;i++)
printf("%d ",ans[i]);
printf("\n");
return 0;
}