題目:
在數字三角形中尋找一條從頂部到底邊的路徑,使得路徑上所經過的數字之和最大。路徑上的每一步都只能往左下或 右下走。只需要求出這個最大和即可,不必給出具體路徑。三角形的行數大於1小於等於100,數字為 0 - 99。
輸入格式:
5 //表示三角形的行數 接下來輸入三角形 7 3 8 8 1 0 2 7 4 4 4 5 2 6 5
要求輸出最大和。
思路分析:
這里的遞歸和記憶型遞歸都很容易理解,遞歸和記憶型遞歸都是自頂向下,動規則是自底向上,由小規模向上推。
這里寫dp方程的時候有一個思路,那就是依賴誰先求誰,這里依賴的是下一步的最大值,於是我們就可以先求出下一步的最大值,然后往上遞推求出我們最終需要的值。我們可以用二維數組存儲從最后一行開始每個路徑的結果。
那有些題目有空間大小的限制,這里就涉及到另一種技巧,滾動數組,滾動數組就是把本身需要保留的結果,但是利用過后直接用下一次的結果給覆蓋掉,這樣的一個數組就是滾動數組。
滾動數組是DP中的一種編程思想。簡單的理解就是讓數組滾動起來,每次都使用固定的幾個存儲空間,來達到壓縮,節省存儲空間的作用。起到優化空間,主要應用在遞推或動態規划中(如01背包問題)。因為DP題目是一個自底向上的擴展過程,我們常常需要用到的是連續的解,前面的解往往可以舍去。所以用滾動數組優化是很有效的。利用滾動數組的話在N很大的情況下可以達到壓縮存儲的作用。
代碼:
import java.time.Duration; import java.time.Instant; import java.util.Scanner; public class 數字三角形 { public static void main(String[] args) { // int[][] triangle = { // {7}, // {3, 8}, // {8, 1, 0}, // {2, 7, 4, 4}, // {4, 5, 2, 6, 5}, // {4, 5, 2, 6, 5, 7}, // {4, 13, 12, 88, 6, 6, 5}, // {3, 8, 7, 11, 9, 22, 66, 3}, // }; // Instant now = Instant.now(); // System.out.println(maxSumUsingRecursive(triangle, 0, 0)); // System.out.println("持續時間為:" + // Duration.ofMillis(Instant.now().toEpochMilli() - // now.toEpochMilli()).getSeconds()); // // // now = Instant.now(); // System.out.println(maxSumUsingMemory(triangle, 0, 0, new int[8][8])); // System.out.println("持續時間為:" + // Duration.ofMillis(Instant.now().toEpochMilli() - // now.toEpochMilli()).getSeconds()); Scanner sc = new Scanner(System.in); int n = sc.nextInt(); int[][] triangle = new int[n][]; for (int i = 0; i < n; i++) { triangle[i] = new int[i + 1]; for (int j = 0; j < i + 1; j++) { triangle[i][j] = sc.nextInt(); } } System.out.println(maxSumUsingDp(triangle, 0, 0)); } /** * @param triangle 數字三角形 * @param i 起點行號 * @param j 起點列號 * @return 計算出的最大和 */ public static int maxSumUsingRecursive(int[][] triangle, int i, int j) { int rowIndex = triangle.length; if (i == rowIndex - 1) { return triangle[i][j]; } else { // 頂點的值+max(左側支路的最大值,右側支路的最大值) return triangle[i][j] + Math.max(maxSumUsingRecursive(triangle, i + 1, j), maxSumUsingRecursive(triangle, i + 1, j + 1)); } } /** * 記憶型遞歸 */ public static int maxSumUsingMemory(int[][] triangle, int i, int j, int[][] map) { int rowIndex = triangle.length; int value = triangle[i][j]; if (i == rowIndex - 1) { } else { // 緩存有值,便不遞歸 int v1 = map[i + 1][j]; if (v1 == 0) { v1 = maxSumUsingMemory(triangle, i + 1, j, map); } // 緩存有值,便不遞歸 int v2 = map[i + 1][j + 1]; if (v2 == 0) { v2 = maxSumUsingMemory(triangle, i + 1, j + 1, map); } value = value + Math.max(v1, v2); } // 放入緩存 map[i][j] = value; return value; } // 滾動數組 public static int maxSumUsingDp(int[][] triangle, int i, int j) { int rowCount = triangle.length;// 行數 int columnCount = triangle[rowCount - 1].length;// 最后一行的列數 int[] dp = new int[columnCount]; for (int k = 0; k < columnCount; k++) { dp[k] = triangle[rowCount - 1][k];// 初始化最后一行 } for (int k = rowCount - 2; k >= 0; k--) { for (int l = 0; l <= k; l++) { dp[l] = triangle[k][l] + Math.max(dp[l], dp[l + 1]); } } return dp[0]; } }
結果:
路徑為:7 3 8 7 5。