教主的花园
题目描述
教主有着一个环形的花园,他想在花园周围均匀地种上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;
}