P1040加分二叉树

据说窝回去的那几天考了n次试

过去了一个月才想起来补锅

传送

首先来区分一下什么是中序遍历,什么又是前序遍历

中序遍历:左,根,右(也就是说给出一个序列(按照中序遍历的序列),第i个点左边的点都是i的左子树,右边的点都是i 的右子树)

前序遍历:根,左,右,也就是我们平常画一棵树的顺序

因为这个题目中只给出了中序遍历的顺序,但是画出来的树可以千奇百怪

比如说:

样例可以长成这个样子

还可以长成这样子

所以树的样子与树根的选择有着密切的联系

上面中序遍历中提到了一个性质:给出一个序列(按照中序遍历的序列),第i个点左边的点都是i的左子树,右边的点都是i 的右子树

仔细观察,发现这道题就是让我们对于每棵子树选出树根,使总得分最大,树根也就是上面的i。

似乎是个区间dp?窝也不知道啊,就当它是吧。

结合上面的性质,我们在枚举k(第三层枚举的断点)的时候,就相当于在枚举树根。考虑到要输出前序遍历,我们就把最终选定的树根记录下来。

我们设f[i][j]表示区间[i,j]的最大得分,root[i,j]表示区间[i,j]中选定的树根。

则:f[i][j]=max{f[i][k-1]*f[k+1][j]+f[k][k]}(i<k<j),初始化:f[i][i]=i,root[i][i]=i,f[i][j]=f[i+1][j]=f[i][i](这里是假定左子树为空,如果左子树不为空,肯定会有更优的解覆盖掉它)

区间dp步骤:

第一层:枚举区间长度(<n)

第二层:枚举起点st(st+len<=n),计算终点end=st+len

第三层:枚举断点k(st<=k<end)

在这道题里面,由于f[st][end]初始化时就是k=st的情况,所以k直接从st+1开始

输出前序遍历:递归输出。首先输出[l,r]的根节点,再输出左子树,再输出右子树。(详情见代码)

代码:

#include<bits/stdc++.h>
#define ll long long
using namespace std;
ll n,f[60][60],root[60][60];//注意数据范围
ll read()
{
   char ch=getchar();
   ll x=0;    bool f=0;
   while(ch<'0'||ch>'9')
   {
        if(ch=='-')
         f=1;
        ch=getchar();
   }
   while(ch>='0'&&ch<='9')
   {
       x=(x<<3)+(x<<1)+(ch^48);
       ch=getchar();
   }
   return f?-x:x;
} 
void print(int l,int r)
{
    if(l>r)return;
    printf("%lld ",root[l][r]);
    if(l==r)return ;//注意顺序
    print(l,root[l][r]-1);//注意这里是按照根节点分左右的
    print(root[l][r]+1,r);
}
int main()
{
    n=read();
    for(int i=1;i<=n;i++)
     f[i][i]=read(),root[i][i]=i;
    for(int len=1;len<n;len++)
    {
        for(int st=1;st+len<=n;st++)
        {
            int end=st+len;
            f[st][end]=f[st+1][end]+f[st][st];
            root[st][end]=st;
            for(int k=st+1;k<end;k++)
            {
                if(f[st][end]<f[st][k-1]*f[k+1][end]+f[k][k])
                  f[st][end]=f[st][k-1]*f[k+1][end]+f[k][k],root[st][end]=k;
            }
        }
    }
    printf("%lld\n",f[1][n]);
    print(1,n);
}

 

posted @ 2019-07-04 09:28  千载煜  阅读(295)  评论(0编辑  收藏  举报