codevs 1048 石子归并(区间型DP)

题目描述 Description
有n堆石子排成一列,每堆石子有一个重量w[i], 每次合并可以合并相邻的两堆石子,一次合并的代价为两堆石子的重量和w[i]+w[i+1]。问安排怎样的合并顺序,能够使得总合并代价达到最小。

输入描述 Input Description
第一行一个整数n(n<=100)
第二行n个整数w1,w2…wn (wi <= 100)

输出描述 Output Description
一个整数表示最小合并代价

样例输入 Sample Input
4
4 1 1 4

样例输出 Sample Output
18

题解:一道动态规划的题目,对于这个题我们定义dp[i][j]为合并区间(i,j)的代价,所以状态转移方程为dp[i][j]=min{dp[i][k]+dp[k+1][j]+sum[i][j]}(i<=k<=j-1),sum[i][j]为区间(i,j)的代价和,可以用前缀和优化掉,所以总复杂度是O(n^3)。这道题还需要初始化dp数组为极大值,且dp[i][i]=0(1<=i<=n)。

代码如下

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
int a[10100],sum[10100];
int dp[101][101];
int main()
{
    int n;
    scanf("%d",&n);
    memset(dp,63,sizeof(dp));//初始化 
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        sum[i]=sum[i-1]+a[i];
        dp[i][i]=0;//显然合并(i,i)的代价为0 
    }
    for(int i=n;i>=1;i--)
        for(int j=i+1;j<=n;j++)
            for(int k=i;k<j;k++)
                dp[i][j]=min(dp[i][j],dp[i][k]+dp[k+1][j]+sum[j]-sum[i-1]);
    printf("%d",dp[1][n]);//输出合并区间(1,n)的代价 
    return 0;
}
发布了81 篇原创文章 · 获赞 2 · 访问量 4万+

猜你喜欢

转载自blog.csdn.net/Loi_YZS/article/details/52982771
今日推荐