题目:http://dev.luogu.org:3308/problem/show?pid=1040
又一道区间DP的好题。
枚举区间,对这一区间的分数进行判断(感谢善良的出题人已经把节点自动变成了1..n的中序遍历),判断是否大于原分数,如果大于的话记下目前的根节点和最大分数
预处理时:
f[i][i]=num[i]
f[i][j]=1(i!=j)
root[1][i]=1(root即这段区间的根节点)
转移方程:
f[i][j]=max(f[i][k-1]*f[k+1][j]+f[k][k])
代码如下:
#include <stdio.h>
#include <math.h>
using namespace std;
#define rep(i,k,n) for(int i=k;i<=n;i++)
#define repu(i,n,k) for(int i=n;i>=k;i--)
typedef long long ll;
int n,m,root[40][40],i,j,k;
ll f[40][40];
void tree(int l,int r)//将得到的二叉树输出
{
if (l<=r)
{
printf("%d ",root[l][r]);
tree(l,root[l][r]-1);
tree(root[l][r]+1,r);
}
}
int main()
{
scanf("%d",&n);
rep(i,0,n)
rep(j,0,n)
{
f[i][j]=1;
root[i][i]=i;
}
rep(i,1,n)
scanf("%lld",&f[i][i]);
repu(i,n,1)
rep(j,i+1,n)
rep(k,i,j)
{
if (f[i][j]<f[i][k-1]*f[k+1][j]+f[k][k])
{
f[i][j]=f[i][k-1]*f[k+1][j]+f[k][k];
root[i][j]=k;
}
}
printf("%lld\n",f[1][n]);
tree(1,n);
return 0;
}