开启掘金成长之旅!这是我参与「掘金日新计划 · 2 月更文挑战」的第 11 天,点击查看活动详情
题目:教主有着一个环形的花园,他想在花园周围均匀地种上n棵树,但是教主花园的土壤很特别,每个位置适合种的树都不一样,一些树可能会因为不适合这个位置的土壤而损失观赏价值。 教主最喜欢33种树,这3种树的高度分别为10,20,30。教主希望这一圈树种得有层次感,所以任何一个位置的树要比它相邻的两棵树的高度都高或者都低,并且在此条件下,教主想要你设计出一套方案,使得观赏价值之和最高。
思路:f[i][j][0/1]代表第i个种第j棵树比周围树低/高所能获得最大的价值,因为是环形所以把第一个位置单独拿出来,枚举第一个位置种第几棵树,然后从第二个位置开始转移,转移方程也很好想
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][0][0]=max(f[i-1][1][1],f[i-1][2][1])+a[i][0];
f[i][2][1]=max(f[i-1][1][0],f[i-1][0][0])+a[i][2];
之后就是代码了
#include<bits/stdc++.h>
//#pragma-GCC-optimize("-Ofast");
#define ll long long
#define int long long
#define lowbit(x) ((x)&(-x))
#define endl '\n'
using namespace std;
const ll mod=998244353;
const ll inf=1e9;
const double pi=acos(-1);
const int N=1e6+100;
ll qpow(ll a,ll b)
{
ll res=1;
while(b)
{
if(b&1) res=res*a%mod;
a=a*a%mod;
b>>=1;
}
return res;
}
ll getinv(ll a){return qpow(a,mod-2);}
int a[100005][3],f[100005][3][3],n;
signed main()
{
//ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
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][1][0]=f[i-1][2][1]+a[i][1];
f[i][1][1]=f[i-1][0][0]+a[i][1];
f[i][0][0]=max(f[i-1][1][1],f[i-1][2][1])+a[i][0];
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<<endl;
return 0;
}