题目描述
设有 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点共走两次,试找出 22 条这样的路径,使得取得的数之和为最大。
输入格式
输入的第一行为一个整数 N(表示 N×N 的方格图),接下来的每行有三个整数,前两个表示位置,第三个数为该位置上所放的数。一行单独的 0 表示输入结束。
输出格式
只需输出一个整数,表示 2 条路径上取得的最大的和。
思路分析:
刚开始想的很简单,想着利用两遍动态规划就ok,之后被卡到了60分,看到别人的题解举出的反例:
5 | ||
2 | 0 | |
0 | 4 |
第一次最优是:0+3+6+4+0=13
2 | 0 | |
0 | 0 |
第二次最优是:0+0+5+0+0=5
两次和应为18.
而事实上是第一次:0+2+6+4+0=12 第二次:0+3+5+0+0=8
两次和应为20.
所以就说明递推的思路找错了,就换个思路,求的是两次从A到B的最大值,分开求会出现问题,所以可以试试一起求,
也就是说用一个四维的变量来表示,例如dp[a][b][c][d],表示(a,b)到B点的数值和加(c,d)到B点的数值和相加的最大值。
而dp[a][b][c][d]是可以由它的下面或者右面得到的。一共有四种可能即下下,下右,右右,右下。
因为要遍历所有的点也就是一个的时间复杂度,最大为9也就是,大概是多,时间上应该是可以满足。
#include<iostream>
#include<algorithm>
using namespace std;
int const N = 10;
int mp[N][N];
int dp[N][N][N][N];//[][]是一个到B的位置 [][]一个点到B的位置
#define max_(a,b,c,d) max(max(max(a,b),c),d)
int findmax(int n) {
for(int i = n - 1; i >= 0; i--){
for(int j = n - 1;j >= 0; j--){
for(int k = n - 1;k >= 0;k--){
for(int l = n - 1;l >= 0;l--){
dp[i][j][k][l] = max_(
dp[i+1][j][k+1][l],
dp[i+1][j][k][l+1],
dp[i][j+1][k][l+1],
dp[i][j+1][k+1][l]
)+mp[i][j]+(i == k && j == l ? 0 :mp[k][l]);
}
}
}
}
return dp[0][0][0][0];
}
int main() {
int n;
cin >> n;
int a, b, c;
while (true) {
cin >> a >> b >> c;
if (a == b && b == c && a == 0) {
break;
}
else {
mp[a - 1][b - 1] = c;
}
}
cout << findmax(n) << endl;
}