1. 程式人生 > 實用技巧 >NC16759 方格取數O(n^3)

NC16759 方格取數O(n^3)

連結:https://ac.nowcoder.com/acm/problem/16759
來源:牛客網

Problem

題目描述

設有N*N的方格圖(N ≤ 10,我們將其中的某些方格中填入正整數,而其他的方格中則放入數字0。如下圖所示(見樣例):

某人從圖的左上角的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

輸出

67

Solution

選最大的兩次不是最優的,所以是dp,本題擴充套件到四維即可,但是可以根據每次轉移的座標和一樣,優化到三維。

Code

#include<stdio.h>
#include<iostream>
using namespace std;
int f[10][2]={{0,1},{0,-1},{1,0},{-1,0}};
template<class T>inline void read(T &res){
	char c;T flag=1;
	while((c=getchar())<'0'||c>'9')if(c=='-')flag=-1;res=c-'0';
	while((c=getchar())>='0'&&c<='9')res=res*10+c-'0';res*=flag;
}
int n,m;
int a[20][20];
int dp[30][20][20]; 
int main(){
	//printf("%d\n",!0);
	scanf("%d",&n);
	int x,y,z;
	while(scanf("%d%d%d",&x,&y,&z)==3&&x+y+z!=0){
		a[x][y]=z;
	} 
	dp[0][0][0]=a[1][1];
	for(int i=1;i<=2*n-2;i++){
		for(int j=0;j<=i;j++){
			for(int k=0;k<=i;k++){
				int t=0;
				if(i-j>0&&i-k>0) t=max(t,dp[i-1][j][k]);
				if(i-k>0&&j>0) t=max(t,dp[i-1][j-1][k]);
				if(i-j>0&&k>0) t=max(t,dp[i-1][j][k-1]);
				if(j>0&&k>0) t=max(t,dp[i-1][j-1][k-1]);
				if(j==k){
					dp[i][j][k]=t+a[j+1][i-j+1];
				}
				else{
					dp[i][j][k]=t+a[j+1][i-j+1]+a[k+1][i-k+1];
				}
			}
		}
	}
	printf("%d\n",dp[2*n-2][n-1][n-1]);
	return 0;
}