算法笔记_135:格子取数问题(Java)
目录
1 问题描述
有n*n个格子,每个格子里有正数或者0,从最左上角往最右下角走,只能向下和向右走,一共走两次(即从左上角往右下角走两趟),把所有经过的格子里的数加起来,求总和的最大值。如果两次经过同一个格子,则最后求得的总和中该格子中的数只加一次。
2 解决方案
此处采用动态规划法,可以较大的提高时间效率。
具体代码如下:
package com.liuzhen.practice; import java.util.Scanner; public class Main { public boolean judge(int s, int i ,int j ,int len) { int x1 = s - i, x2 = s - j; if(x1 >= 0 && x1 < len && x2 >= 0 && x2 < len && i >= 0 && j >= 0 && i < len && j < len) return true; return false; } public int getValue(int[][][] dp, int s, int i , int j, int len) { if(judge(s, i, j, len)) return dp[s][i][j]; return -1; } public void getResult(int[][] value) { int len = value.length; int[][][] dp = new int[len * 2][len][len]; dp[0][0][0] = value[0][0]; for(int s = 1;s <= len * 2 - 2;s++) { for(int i = 0;i < len;i++) { for(int j = 0;j < len;j++) { if(judge(s, i, j, len)) { if(i != j) dp[s][i][j] = Math.max(Math.max(getValue(dp,s-1,i-1,j-1,len),getValue(dp,s-1,i,j,len)), Math.max(getValue(dp,s-1,i-1,j,len), getValue(dp,s-1,i,j-1,len))) + value[i][s-i] + value[j][s-j]; else dp[s][i][j] = Math.max(getValue(dp,s-1,i-1,j-1,len), Math.max(getValue(dp,s-1,i-1,j,len), getValue(dp,s-1,i,j,len))) + value[i][s-i]; } } } } System.out.println(dp[2 * len - 2][len - 1][len - 1]); return; } public static void main(String[] args) { Main test = new Main(); Scanner in = new Scanner(System.in); int n = in.nextInt(); int[][] value = new int[n][n]; for(int i = 0;i < n;i++) for(int j = 0;j < n;j++) value[i][j] = in.nextInt(); test.getResult(value); } }
运行结果:
6 0 0 3 0 2 0 0 0 3 0 0 0 0 0 3 0 0 0 0 0 0 0 4 0 0 0 0 0 4 0 0 0 3 0 0 0 22 8 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 67
参考资料:
1.《编程之法面试和算法心得》 July 著
版权声明:本文为liuzhen1995原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。