vijos 1006 晴天小猪历险记之Hill (环型DP + 数塔)

背景

在很久很久以前，有一个动物村庄，那里是猪的乐园（^_^），村民们勤劳、勇敢、善良、团结……
不过有一天，最小的小小猪生病了，而这种病是极其罕见的，因此大家都没有储存这种药物。所以晴天小猪自告奋勇，要去采取这种药草。于是，晴天小猪的传奇故事便由此展开……

描述

这一天，他来到了一座深山的山脚下，因为只有这座深山中的一位隐者才知道这种药草的所在。但是上山的路错综复杂，由于小小猪的病情，晴天小猪想找一条需时最少的路到达山顶，但现在它一头雾水，所以向你求助。
山用一个三角形表示，从山顶依次向下有1段、2段、3段等山路，每一段用一个数字T（1<=T<=100）表示，代表晴天小猪在这一段山路上需要爬的时间，每一次它都可以朝左、右、左上、右上四个方向走。山是环形的。（注意：在任意一层的第一段也可以走到本层的最后一段或上一层的最后一段）。
晴天小猪从山的左下角出发，目的地为山顶，即隐者的小屋。

格式 输入格式

第一行有一个数n（2<=n<=1000），表示山的高度。
从第二行至第n+1行，第i+1行有i个数，每个数表示晴天小猪在这一段山路上需要爬的时间。
输出格式

一个数，即晴天小猪所需要的最短时间。

样例1 样例输入1[复制]

5
1
2 3
4 5 6
10 1 7 8
1 1 4 5 6

样例输出1[复制]

10
限制
各个测试点1s

题解

难点是当层状态的优化, 因为是一个环, 咋一看不符合DP的无后效性要求. 但可以来回DP一次, 要先只向一个方向DP, 暂时不看另一个方向, 看完一个方向, 再看另一个方向, 这样才能保证无后效性

import java.util.Scanner;

public class Main {

  public static void main(String[] args) {
    Scanner sc = new Scanner(System.in);
    int n = sc.nextInt();
    int[][] arr = new int[n][n];
    for (int i = 0; i < n; ++i) {
      for (int j = 0; j <= i; ++j) {
        arr[i][j] = sc.nextInt();
      }
    }
    int[][] dp = new int[n][n];
    //初始化状态
    dp[n - 1][0] = arr[n - 1][0];
    for (int j = 1; j < n; ++j) { //从左到右
      dp[n - 1][j] = arr[n - 1][j] + dp[n - 1][j - 1];
    }
    dp[n - 1][n - 1] = Math.min(dp[n - 1][n - 1], arr[n - 1][n - 1]) + dp[n - 1][0];
    for (int j = n - 1 - 1; j >= 0; --j) { //从右到左
      dp[n - 1][j] = Math.min(dp[n - 1][j], arr[n - 1][j] + dp[n - 1][j + 1]);
    }

    for (int i = n - 2; i >= 0; --i) {
      //DP下一层状态
      for (int j = 0; j <= i; ++j) {
        if (j == 0) {
          dp[i][j] = arr[i][j] + Min(dp[i + 1][j], dp[i + 1][j + 1], dp[i + 1][i + 1]);
        } else if (j == i) {
          dp[i][j] = arr[i][j] + Min(dp[i + 1][j], dp[i + 1][j + 1], dp[i + 1][0]);
        } else {
          dp[i][j] = arr[i][j] + Math.min(dp[i + 1][j], dp[i + 1][j + 1]);
        }
      }
      //DP优化当层状态, 为下层状态做准备
      if (i != 0) {
        dp[i][0] = Math.min(dp[i][0], arr[i][0] + dp[i][i]);
        for (int j = 1; j <= i; ++j) {
          dp[i][j] = Math.min(dp[i][j], arr[i][j] + dp[i][j - 1]);
        }
        dp[i][i] = Math.min(dp[i][i], arr[i][i] + dp[i][0]);
        for (int j = i - 1; j >= 0; --j) {
          dp[i][j] = Math.min(dp[i][j], arr[i][j] + dp[i][j + 1]);
        }
      }
    }
    System.out.println(dp[0][0]);
  }

  static int Min(int a, int b, int c) {
    if (a < b) {
      a ^= b ^= a ^= b;
    }
    if (b < c) {
      b ^= c ^= b ^= c;
    }
    return c;
  }
}