动态规划解传纸条问题
题目描述
小渊和小轩是好朋友也是同班同学,他们在一起总有谈不完的话题。一次素质拓展活动中,班上同学安排做成一个 m行 n 列的矩阵,而小渊和小轩被安排在矩阵对角线的两端,因此,他们就无法直接交谈了。幸运的是,他们可以通过传纸条来进行交流。纸条要经由许多同学传到对方手里,小渊坐在矩阵的左上角,坐标 (1,1),小轩坐在矩阵的右下角,坐标 (m,n)。从小渊传到小轩的纸条只可以向下或者向右传递,从小轩传给小渊的纸条只可以向上或者向左传递。
在活动进行中,小渊希望给小轩传递一张纸条,同时希望小轩给他回复。班里每个同学都可以帮他们传递,但只会帮他们一次,也就是说如果此人在小渊递给小轩纸条的时候帮忙,那么在小轩递给小渊的时候就不会再帮忙。反之亦然。
还有一件事情需要注意,全班每个同学愿意帮忙的好感度有高有低(注意:小渊和小轩的好心程度没有定义,输入时用 0 表示),可以用一个[0,100] 内的自然数来表示,数越大表示越好心。小渊和小轩希望尽可能找好心程度高的同学来帮忙传纸条,即找到来回两条传递路径,使得这两条路径上同学的好心程度之和最大。现在,请你帮助小渊和小轩找到这样的两条路径。
输入格式
第一行有两个用空格隔开的整数 m 和 n,表示班里有 m行 n 列。
接下来的 m行是一个 m×n的矩阵,矩阵中第 i 行 j列的整数表示坐在第 i行 j列的学生的好心程度。每行的 n个整数之间用空格隔开。
输出格式
输出文件共一行一个整数,表示来回两条路上参与传递纸条的学生的好心程度之和的最大值。
输入输出样例
输入 #1
3 3
0 3 9
2 8 5
5 7 0
输出 #1复制
34
解题思路
两个人互相传纸条,可以理解为从一个人以不同的路径向另一个人传两张纸条。也就是说,我们只需要找到从起点到终点的两条不想交路径,使好感度之和最大。
用动态规划,我们需要确定目标状态和转移方程。
由于只能横向或纵向移动并且终点值为0,所以目标状态就是终点上面和左面的两个点,即f[m][n-1][m-1][n]
转移方程是f[i][j][k][l]=``max(f[i-1][j][k-1][l],f[i-1][j][k][l-1],f[i][j-1][k][l-1],f[i][j-1][k-1][l])+a[i][j]+a[k][l]
完整代码
#include <iostream>
using namespace std;
int a[100][100];
int f[100][100][100][100];
int max(int a,int b,int c,int d)//四个数中取最大
{
if(b>a) a=b;
if(c>a) a=c;
if(d>a) a=d;
return a;
}
int main()
{
int m,n;
cin>>m>>n;
for(int i=1;i<=m;i++)
for(int j=1;j<=n;j++)
cin>>a[i][j];
//转移方程,找到最优解
for(int i=1;i<=m;i++)
for(int j=1;j<=n;j++)
for(int k=1;k<=m;k++)
for(int l=j+1;l<=n;l++)
f[i][j][k][l]=max(f[i-1][j][k-1][l],f[i-1][j][k][l-1],f[i][j-1][k][l-1],f[i][j-1][k-1][l])+a[i][j]+a[k][l];//转移方程
cout<<f[m][n-1][m-1][n];//目标状态
return 0;
}
另外,如果提交到答题网站不能全部通过,就把数组空间开小一点,55就够了。