【刷题】动态规划——线性DP(数字三角形)：方格取数

题目链接

和标准数字三角形不同的是有两条路径，第一条路径取过的数，第二条路径不能再取。
题目的走两次可以看作：两个人同时走，且不取相同格子的数字。

考虑用4维数组f[i1][j1][i2][j2]存储状态，w[x][y]存储地图。
(i1, j1)表示第一个人的位置坐标
(i2, j2)表示第二个人的位置坐标
考虑f[i1][j1][i2][j2]的上一个状态，两人可能有4种走法

人1	人2
下	下
下	右
右	下
右	右

以(下,下)为例：
1、若俩人走后位置相同，即i1=j2 && j1=j2
f[i1][j1][i2][j2] = f[i1-1][j1][i2-1][j2] + w[i1][j1]
2、若俩人走后位置不同，即i1!=j2 || j1!=j2
f[i1][j1][i2][j2] = f[i1-1][j1][i2-1][j2] + w[i1][j1] + w[i2][j2]

因为是同时走，每次只走一格，有i1 + j1 = i2 + j2
令k = i1 + j1 = i2 + j2，有j1 = k - i1, j2 = k - i1
因此4维状态可以节省成3维状态f[k][i1][i2]
以(下,下)为例，递推式可化简为
1、若俩人走后位置相同，即i1=j2 && j1=j2
f[k][i1][i2] = f[k-1][i1-1][i2-1] + w[i1][k-i1]
2、若俩人走后位置不同，即i1!=j2 || j1!=j2
f[k][i1][i2] = f[k-1][i1-1][i2-1] + w[i1][k-i1] + w[i2][k-i2]

#include <iostream>
using namespace std;
int n;
int f[25][15][15], w[15][15];

int main() {
    scanf("%d", &n);
    int x, y;
    while(cin >> x >> y >> w[x][y] && (x || y || w[x][y]))
    for(int k = 2; k <= 2*n; k++) {
    	for(int i1 = 1; i1 <= n; i1++) {
	        for(int i2 = 1; i2 <= n; i2++) {
	        	if (k-i1 >= 1 && k-i1 <= n && k-i2 >= 1 && k-i2 <=n) 
				{
					int sum = 0;
	        		if (i1 == i2 && k-i1 == k-i2) {
	        			sum = w[i1][k-i1];
					}
		        	else {
		        		sum = w[i1][k-i1] + w[i2][k-i2];
					}
					int d_d = f[k-1][i1-1][i2-1];
	        		int d_r = f[k-1][i1-1][i2];
	        		int r_d = f[k-1][i1][i2-1];
	        		int r_r = f[k-1][i1][i2];
	        		f[k][i1][i2] = max(d_d, max(d_r, max(r_d, r_r))) + sum;
				}
	        }
	    }
	}
    printf("%d\n", f[2*n][n][n]);
    return 0;
}