石子合并(区间dp+记忆化搜索)

经典例题:石子合并 题目链接
N 堆石子排成一行,现要将石子有次序地合并成一堆,规定每次只能选相邻的2堆合并成新的一堆,并将新的一堆的石子数,记为该次合并的得分。计算合并最小得分。

方法一、区间dp

#include<bits/stdc++.h>
using namespace std;

const int N = 1010;
int a[N], sum[N], n, dp[N][N];

signed main(){
    ios::sync_with_stdio(false);
    cin.tie(0); cout.tie(0);
    cin >> n;
    for(int i = 1; i <= n; i++){
        cin >> a[i];
        sum[i] = sum[i - 1] + a[i];
    }
    //从枚举小区间开始
    for(int len = 1; len <= n; len ++){
        for(int i = 1; i + len <= n; i ++){
            int j = i + len;
            int ans = 1e9 + 10;
            for(int k = i; k < j; k ++){
                 ans = min(ans, dp[i][k] + dp[k + 1][j] + sum[j] - sum[i - 1]);
                 dp[i][j] = ans;
            }
        }
    }
    cout << dp[1][n];
    return 0;
}

方法二、记忆化搜索

#include<bits/stdc++.h>
using namespace std;

const int N = 1010;
int n, a[N], dp[N][N], sum[N];
bool vis[N][N];

int dfs(int i, int j){
    if(i == j) return 0;
    if(vis[i][j]) return dp[i][j];
    vis[i][j] = true;
    int ans = 1e9 + 10;
    for(int k = i; k < j; k ++){
        ans = min(ans, dfs(i, k) + dfs(k + 1, j) + sum[j] - sum[ i - 1]);
    }
    dp[i][j] = ans;
    return dp[i][j];
}

signed main(){
    ios::sync_with_stdio(false);
    cin.tie(0); cout.tie(0);
    cin >> n;
    for(int i = 1; i <= n; i++){
        cin >> a[i];
        sum[i] = sum[i - 1] + a[i];//记录前缀和
    }
    cout << dfs(1, n);
    return 0;
}
posted @ 2023-02-20 20:13  风归去  阅读(40)  评论(0编辑  收藏  举报