动态规划解方格取数问题

动态规划解方格取数问题

题目描述

设有 N×N的方格图 (N≤9)，我们将其中的某些方格中填入正整数，而其他的方格中则放入数字 0。如下图所示（见样例）:

A
 0  0  0  0  0  0  0  0
 0  0 13  0  0  6  0  0
 0  0  0  0  7  0  0  0
 0  0  0 14  0  0  0  0
 0 21  0  0  0  4  0  0
 0  0 15  0  0  0  0  0
 0 14  0  0  0  0  0  0
 0  0  0  0  0  0  0  0
                         B

某人从图的左上角的 A 点出发，可以向下行走，也可以向右走，直到到达右下角的 B 点。在走过的路上，他可以取走方格中的数（取走后的方格中将变为数字 0）。
此人从 A 点到 B 点共走两次，试找出 2 条这样的路径，使得取得的数之和为最大。

输入格式

输入的第一行为一个整数 N（表示 N×N的方格图），接下来的每行有三个整数，前两个表示位置，第三个数为该位置上所放的数。一行单独的 0 表示输入结束。

输出格式

只需输出一个整数，表示 2 条路径上取得的最大的和。

输入输出样例

输入 #1

8
2 3 13
2 6  6
3 5  7
4 4 14
5 2 21
5 6  4
6 3 15
7 2 14
0 0  0

输出 #1

67

解题思路

这个问题和传纸条类似，也比较适合用动态规划求解。用动态规划我们需要找到目标状态和转移方程。

我们用f[i][j][k][l]表示第一条路线横纵坐标i,j，第二条路线横纵坐标k,l。

那么目标状态很明显就是f[n][n][n][n];

如果用num[i][j]表示对应位置的数字，那么转移方程就是：f[i][j][k][l]=max(f[i-1][j][k-1][l],f[i][j-1][k-1][l],f[i-1][j][k][l-1],f[i][j-1][k][l-1])+num[i][j]+num[k][l];

这里需要注意，因为我们没有把取走数字的位置赋为0，所以需要在两条路线相交的位置把重复加了两次的值减去一次，也就是加一个如下的判断：if((i==k)&&(j==l)) f[i][j][k][l]=f[i][j][k][l]-num[i][j];

完整代码

#include <iostream>
using namespace std;

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 n;
  	cin>>n;
  	int num[12][12]={0},f[12][12][12][12];
  	while(1)
  	  {
  	  	int a,b,c;
  	  	cin>>a>>b>>c;
  	  	if((a==0)&&(b==0)&&(c==0)) break;
  	  	else num[a][b]=c;
	  }
	int i,j,k,l;
	for(i=1;i<=n;i++)
	  {for(j=1;j<=n;j++)
	     {for(k=1;k<=n;k++)
	        {for(l=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-1][l],f[i][j-1][k][l-1])+num[i][j]+num[k][l];
	            if((i==k)&&(j==l)) f[i][j][k][l]=f[i][j][k][l]-num[i][j];
			   } 
			}
		 }
	  }
	cout<<f[n][n][n][n];
	return 0;
  }