寝癖頭の解法

学習中の覚え書きを投稿、更新していきます。

Aizu Online Judge in C++ #ALDS1_10_D : Optimal Binary Search Tree

Aizu Online Judge(AOJ)の過去問から、その提出コードの解答例です。

・問題 "Optimal Binary Search Tree"
https://onlinejudge.u-aizu.ac.jp/problems/ALDS1_10_D
・最適二分探索木

僕が作成、提出したコードは、以下のとおりです。

Aizu Online Judge in C++ #ALDS1_10_D : Optimal Binary Search Tree
/*
Aizu Online Judge in C++ #ALDS1_10_D : Optimal Binary Search Tree
https://onlinejudge.u-aizu.ac.jp/problems/ALDS1_10_D
 提出コードの解答例
 https://neguse-atama.hatenablog.com
*/
#include<bits/stdc++.h>
using namespace std;
using ll=long long;
double dp[510][510];
int main(void){
    ll n;
    cin>>n;
    double p[510],q[510];
    for(ll i=0;i<n;i++){
        cin>>p[i];
    }
    for(ll i=0;i<=n;i++){
        cin>>q[i];
    }
    for(ll w=0;w<=n;w++){
        for(ll i=0;i<=n-w;i++){
            ll j=i+w;
            dp[i][j]=1e9+7;
            double sum=q[j];
            if(w==0){
                dp[i][j]=sum;
                continue;
            }
            for(ll k=i;k<j;k++){
                sum+=p[k]+q[k];
                dp[i][j]=min(dp[i][j],dp[i][k]+dp[k+1][j]);
            }
            dp[i][j]+=sum;
        }
    }
    cout<<dp[0][n]<<endl;
    return 0;
}

設問の出典は、プログラミング問題のオンライン採点システム「Aizu Online Judge(AOJ)」です。
http://judge.u-aizu.ac.jp/onlinejudge/