NOIP 2003 加分二叉树

评测传送

这个题可以记忆化搜索做的。
注意边界条件。
对于先序遍历，递归就可以了（在dfs的时候用一个数组记录根节点就可以了）。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<ctime>
#include<queue>
#define LL long long
using namespace std;
int n;
LL a[50],ans,sum;
int way[50][50];
LL f[50][50];
LL dfs(int l,int r)
{
    if(l==r) return a[l];
    if(r<l) return 1;
    LL tot=0;
    for(int i=l;i<=r;i++)
    {
        LL s1=f[l][i-1]?f[l][i-1]:dfs(l,i-1);
        LL s2=f[i+1][r]?f[i+1][r]:dfs(i+1,r);
        if(s1*s2+a[i]>tot)
        {
            tot=s1*s2+a[i];
            way[l][r]=i;
        }
    }
    return f[l][r]=tot;
}
void Print(int l,int r)
{
    if(l>=r) 
    {
        if(l==r) printf("%d ",l);
        return;
    }

    printf("%d ",way[l][r]);
    Print(l,way[l][r]-1);
    Print(way[l][r]+1,r);
}
int main()
{
    //freopen("tree.in","r",stdin);
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%lld",&a[i]);
    ans=dfs(1,n);
    printf("%lld\n",ans);
    Print(1,n);
    return 0;
}