洛谷 P1040 加分二叉树

P1040 加分二叉树

树形dp，用记忆化搜索即可

//树形dp P1040 //http://www.cnblogs.com/mhpp/p/6628528.html #include<cstdio>#include<algorithm>using namespace std;int ans[31][31];//从l到r的结点构成的子树的最高加分int n;int root[31][31];//记录从l到r的结点构成的子树最高加分时的根结点int a[31];//显然，取最高得分就是取一个合适的根结点，使左子树的加分×右子树的加分＋根的分数最大。  //因此，循环取l到r为根结点，然后对于每种根结点取法，分别递归求出左右子树最高加分，求出此情况最高得分，最终找到合适根结点。  int dfs(int l,int r)//从l到r的结点构成的子树的最高加分 {if(l>r)return 1;//空子树，返回1if(l==r)//只有1个结点的子树，设定root后返回a[l]{//ans[l][r]=a[l];//可以去掉，因为不再用到这个值 root[l][r]=l;//曾经忘记这一行，要注意 return a[l];}if(ans[l][r])return ans[l][r];//记忆化，已经算出答案，直接返回int i,t;for(i=l;i<=r;i++){t=dfs(l,i-1)*dfs(i+1,r)+a[i];if(t>ans[l][r]){root[l][r]=i;ans[l][r]=t;}}return ans[l][r];}void print(int l,int r){if(root[l][r]){printf("%d ",root[l][r]);print(l,root[l][r]-1);print(root[l][r]+1,r);}}int main(){int i;scanf("%d",&n);for(i=1;i<=n;i++)scanf("%d",&a[i]);dfs(1,n);printf("%d\n",ans[1][n]);print(1,n);return 0;}