1. 程式人生 > 實用技巧 >poj 3176 dp

poj 3176 dp

地址http://poj.org/problem?id=3176

大致題意:

輸入一個n層的三角形,第i層有i個數,求從第1層到第n層的所有路線中,權值之和最大的路線。

規定:第i層的某個數只能連線走到第i+1層中與它位置相鄰的兩個數中的一個。

數值範圍 0~350

Sample Input
5
7
3 8
8 1 0
2 7 4 4
4 5 2 6 5
Sample Output
30

樣例路線

Hint

Explanation of the sample:

          7

         *

        3   8

       *

      8   1   0

       *

    2
7 4 4 * 4 5 2 6 5

解答

dp方程 當前所在座標i j 的最大值是上一層兩個元素中的較大值 然後加上自己所在座標的值

dp[i][j] = max(dp[i - 1][j] + arr[i][j], dp[i - 1][j - 1] + arr[i][j]);

起始dp[0][0] = 三角形[0][0];

// poj3176.cpp : 此檔案包含 "main" 函式。程式執行將在此處開始並結束。
//

#include <iostream>
#include <algorithm>



using namespace
std; /* Sample Input 5 7 3 8 8 1 0 2 7 4 4 4 5 2 6 5 Sample Output 30 */ const int N = 500; int arr[N][N]; int dp[N][N]; int n; int main() { cin >> n; for (int i = 0; i < n; i++) { for (int j = 0; j <= i; j++) { cin >> arr[i][j]; } } dp[
0][0] = arr[0][0]; for (int i = 1; i < n; i++) { for (int j = 0; j <= i; j++) { if (j == 0) dp[i][j] = dp[i - 1][j]+arr[i][j]; else if (j == i) dp[i][j] = dp[i - 1][j - 1]+arr[i][j]; else { dp[i][j] = max(dp[i - 1][j] + arr[i][j], dp[i - 1][j - 1] + arr[i][j]); } } } int ans = -9999999; for (int i = 0; i < n; i++) { ans = max(ans, dp[n - 1][i]); } cout << ans << endl; }