教主的花园（动态规划）

教主的花园

题目描述

教主有着一个环形的花园，他想在花园周围均匀地种上n棵树，但是教主花园的土壤很特别，每个位置适合种的树都不一样，一些树可能会因为不适合这个位置的土壤而损失观赏价值。

教主最喜欢3种树，这3种树的高度分别为 10,20,30。教主希望这一圈树种得有层次感，所以任何一个位置的树要比它相邻的两棵树的高度都高或者都低，并且在此条件下，教主想要你设计出一套方案，使得观赏价值之和最高。

题目分析

此题采用动态规划的思想进行解题

设定集合f[i,j,k]表示第i个位置种第j种树且比其两边的树(k=1)高（k=0低）的观赏价值的集合，目的是求种下树后最大的观赏价值

因树的种类较少我们可以分开讨论，以下用w代替应当累加的观赏值

• 当第i树种第一种树时：
  • 因其高度最小，则其两侧的树一定比它高，所以第i-1棵树可能为第二种也可能为第三种，但其状态应为比两侧的树高
  • 可得方程为：f[i,0,0] = max(f[i -1,1,1], f[i - 1,2,1]) + w;
• 当第i树种第二种树时：
  • 则其两侧的树可能比它高也可能比它矮，所以存在两种情况
  • 当第i-1棵树高时，其状态一定为1且为第三种树则有：f[i][1][0] = f[i - 1][2][1] + w;
  • 当第i-1棵树矮时，其状态一定为0且为第一种树则有：f[i][1][1] = f[i - 1][0][0] + w;
• 当第i树种第三种树时
  • 则其两侧的树一定比它矮，所以第i-1棵树可能为第一种也可能为第二种，但其状态应为比两侧的树矮
  • 可得方程：f[i][2][1] = max(f[i - 1][1][0], f[i - 1][0][0]) + w;

还需要注意，既然花园是环形的，第一棵树和最后一颗树就算是相邻的，此两种树也不可以时同一种树我们可以通过枚举第一棵树时那种树，再进行操作，随时更新极大值即可。

code

#include<bits/stdc++.h>

using namespace std;

const int N = 1e5 + 10;

int n, m, k, t;
int f[N][4][3];
int a[N][4];

int main()
{
    cin >> n;
    for(int i = 1; i <= n; i ++) cin >> a[i][0] >> a[i][1] >> a[i][2];

    int ans = 0;
    for(int j = 0; j < 3; j ++)//因环形，第一棵树应当特殊处理
    {
        for(int i = 0; i < 3; i ++)
            for(int k = 0; k < 2; k ++)
                f[1][i][k] = 0;

        f[1][j][0] = f[1][j][1] = a[1][j];
        for(int i = 2; i <= n; i ++)
        {
            f[i][0][0] = max(f[i -1][1][1], f[i - 1][2][1]) + a[i][0];

            f[i][1][0] = f[i - 1][2][1] + a[i][1];
            f[i][1][1] = f[i - 1][0][0] + a[i][1];

            f[i][2][1] = max(f[i - 1][1][0], f[i - 1][0][0]) + a[i][2];
        }

        //最后一颗树不可以和第一棵树相同，分段取最大值
        for(int i = 0; i < j; i ++) ans=max(ans, f[n][i][0]);
        for(int i = 2; i > j; i --) ans = max(ans, f[n][i][1]);
    }

    cout << ans << "\n";

    return 0;
}