sjtu 1077 加分二叉树

树型DP入门题

题目链接：http://acm.sjtu.edu.cn/OnlineJudge/problem/1077

•设f(i,j)中序遍历为i,i+1,…,j的二叉树的最大加分，则有：
f(i,j)=max{f[i,k-1]*f[k+1,j] +d[k]}

•显然 f(i,i)=d[i]
•答案为f(1,n)
•1<=i<=k=<=j<=n
•时间复杂度 O(n3)
•要构造这个树，只需记录每次的决策值，令b(i,j)=k，表示中序遍历为i,i+1,…,j的二叉树的取最优决策时的根结点为k
最后前序遍历这个树即可。

/*
树型DP
*/
#include <cstdio>
#include <cstring>
#include <queue>
using namespace std;
#define MAX 35
#define INF 0x3f3f3f3f

long long dp[MAX][MAX];
int p[MAX][MAX];
int a[MAX];
queue<int>q;

long long dfs(int i ,int j)
{
if(i>j) return dp[i][j]=1;
if(dp[i][j]!=-1) return dp[i][j];
dp[i][j]=-INF;
for(int k=i; k<=j; k++)
{
long long t1=dfs(i,k-1);
long long t2=dfs(k+1,j);
if(t1*t2+a[k] > dp[i][j])
{
dp[i][j]=t1*t2+a[k];
p[i][j]=k;
}
}
return dp[i][j];
}

void travel(int i ,int j)
{
if(i>j) return ;
if(i==j)
{
q.push(i);
return ;
}
int k=p[i][j];
q.push(k);
travel(i,k-1);
travel(k+1,j);
}

int main()
{
int n;
while(scanf("%d",&n)!=EOF)
{
for(int i=1; i<=n; i++) scanf("%d",&a[i]);
memset(p,-1,sizeof(p));
memset(dp,-1,sizeof(dp));
for(int i=1; i<=n; i++)
{
dp[i][i]=a[i];
p[i][i]=i;
}
dfs(1,n);
printf("%lld\n",dp[1]
);
while(!q.empty()) q.pop();
travel(1,n);
printf("%d",q.front());
q.pop();
while(!q.empty())
{
printf(" %d",q.front());
q.pop();
}
printf("\n");
}
return 0;
}