Vijos 1100 (区间DP)

题目链接: https://vijos.org/p/1100

题目大意：NOIP著名的加分二叉树。给出一棵树的中序遍历，加分规则左子树*右子树+根。空子树分数为1。问最大加分的树结构，输出树结构的先序遍历。

解题思路：

先从小的问题看起。

对于一棵子树，只要知道根是啥，就能轻松求出这棵子树的加分情况。

那么就变成枚举根的区间DP问题。

由于要输出先序遍历，则用m[i][j]记录在i~j区间选择的根。

 

区间DP边界：

①一个点情况：即无左右子树，dp[i][i]=node[i],m[i][i]=i.

②两个点情况，即无右子树。dp[i][i+1]=node[i]+node[i+1];m[i][i+1]=i。

注意为什么DP边界是两种情况，是因为区间DP枚举中间分割点时，是按照常规处理左区间和右区间的，以上两种情况，左右区间都是有问题的。

所以需要特别预处理。

 

区间DP：

推荐先枚举区间间隔p的写法，这里直接从p=2开始计算。p=0，p=1已经预处理。

对于dp[i][j]，则根k的范围(i+1,j)，按照规则写方程就行了。m[i][j]=k，记录每个区间的根。

则最后ans=dp[1][n]。

递归打印方案，先序遍历是根左右，不要打错了。

 

#include "cstring"

#include "cstdio"

#include "cstring"

int dp[31][31],m[31][31],node[31];

void print(int i,int j)

{

    printf("%d ",m[i][j]);

    if(i<=m[i][j]-1) print(i,m[i][j]-1);

    if(m[i][j]+1<=j) print(m[i][j]+1,j);

}

int main()

{

    //freopen("in.txt","r",stdin);

    int n;

    while(scanf("%d",&n)!=EOF)

    {

        memset(dp,0,sizeof(dp));

        for(int i=1; i<=n; i++) scanf("%d",&node[i]);

        for(int i=1; i<=n; i++)

        {

            dp[i][i]=node[i];

            m[i][i]=i;

            dp[i][i+1]=node[i]+node[i+1];

            m[i][i+1]=i;

        }

        for(int p=2; p<=n; p++)

        {

            for(int i=1; i<=n; i++)

            {

                int j=i+p;

                if(j>n) break;

                dp[i][j]=1;

                for(int k=i+1; k<j; k++)

                {

                    if((dp[i][k-1]*dp[k+1][j]+node[k])>dp[i][j])

                    {

                        dp[i][j]=dp[i][k-1]*dp[k+1][j]+node[k];

                        m[i][j]=k;

                    }

                }

            }

        }

        printf("%d\n",dp[1][n]);

        print(1,n);

    }

}