这里两道题基本一模一样。
能量项链
思路:(同矩阵链乘)
状态定义,区间dp定义一般都是dp[i][j],将[L,R]的区间合并的方式。
分界线即为l+1~r-1,因为是矩阵链乘,所以转移分dp[i][k]+dp[k][j]
tips:给的数据为n,但由于是矩阵,最后会回到a[1]处,所以长度枚举需要到n+1。
#include <iostream>
#include <algorithm>
using namespace std;
const int N=300,INF=0x3f3f3f3f;
int dp[N][N];
int n;
int a[N];
int main()
{
cin>>n;
for(int i=1;i<=n;i++)
{
cin>>a[i];
a[i+n]=a[i];
}
for(int len=3;len<=n+1;len++)
for(int i=1,j;i+len-1<=n*2;i++)
{
j=i+len-1;
dp[i][j]=-INF;
for(int k=i+1;k<j;k++)
dp[i][j]=max(dp[i][j],dp[i][k]+dp[k][j]+a[i]*a[k]*a[j]);
}
int res=-INF;
for(int i=1;i<=n;i++) res=max(res,dp[i][i+n]);
cout<<res<<endl;
return 0;
}
Eat Walnuts
思路:基本和上题一样。
#include <iostream>
#include <algorithm>
#include <cstdio>
using namespace std;
const int N=110;
int dp[N][N];
int n;
int a[N];
int main()
{
while(~scanf("%d",&n))
{
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
for(int len=3;len<=n;len++)
for(int i=1;i+len-1<=n;i++)
{
int j=i+len-1;
dp[i][j]=0x3f3f3f3f;
for(int k=i+1;k<j;k++)
dp[i][j]=min(dp[i][j],dp[i][k]+dp[k][j]+(a[i]+a[j]+a[k])*(a[i]+a[j]+a[k]));
}
printf("%d\n",dp[1][n]);
}
return 0;
}