首页 > 代码库 > uva 10304 - Optimal Binary Search Tree 区间dp

uva 10304 - Optimal Binary Search Tree 区间dp

题目链接

 

给n个数, 这n个数的值是从小到大的, 给出个n个数的出现次数。 然后用他们组成一个bst。访问每一个数的代价是这个点的深度*这个点访问的次数。 问你代价最小值是多少。

区间dp的时候, 如果l >= r, 那么返回0, l == r-1, 返回两个数中小的一个。 其他情况的话枚举分界点进行状态转移。

#include <bits/stdc++.h>using namespace std;#define mem1(a) memset(a, -1, sizeof(a))const int inf = 1061109567;int dp[260][260], a[260], pre[260];int dfs(int l, int r){    if(~dp[l][r])        return dp[l][r];    if(l >= r)        return dp[l][r] = 0;    if(l == r - 1) {        return dp[l][r] = min(a[l], a[r]);    }    dp[l][r] = inf;    for(int i = l-1; i < r; i++) {        dp[l][r] = min(dp[l][r], dfs(l, i) + dfs(i + 2, r) + pre[r]-pre[l-1] - a[i+1]);    }    return dp[l][r];}int main(){    int n;    while(~scanf("%d", &n)) {        mem1(dp);        for(int i = 1; i <= n; i++) {            scanf("%d", &a[i]);            pre[i] = pre[i-1] + a[i];        }        printf("%d\n", dfs(1, n));    }    return 0;}

 

uva 10304 - Optimal Binary Search Tree 区间dp