首页 > 代码库 > Vijos 1100 (区间DP)

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);    }}

 

Vijos 1100 (区间DP)