码迷,mamicode.com
首页 > 其他好文 > 详细

NYOJ 737 石子合并(一) (区间DP+平行四边形优化)

时间:2015-04-24 21:12:14      阅读:179      评论:0      收藏:0      [点我收藏+]

标签:acm   算法   编程   dp   

题目地址:NYOJ 737
定义状态dp [ i ] [ j ]为从第i个石子到第j个石子的合并最小代价。
没有优化的代码如下:耗时248ms。

#include <iostream>
#include <string.h>
#include <math.h>
#include <queue>
#include <algorithm>
#include <stdlib.h>
#include <map>
#include <set>
#include <stdio.h>
using namespace std;
#define LL long long
#define pi acos(-1.0)
//#pragma comment(linker, "/STACK:1024000000")
const int mod=1e9+7;
const int INF=0x3f3f3f3f;
const double eqs=1e-3;
const int MAXN=40000+10;
int dp[300][300], sum[300];
int main()
{
        int n, i, j, k, len, x;
        while(scanf("%d",&n)!=EOF) {
                sum[0]=0;
                memset(dp,INF,sizeof(dp));
                for(i=1; i<=n; i++) {
                        scanf("%d",&x);
                        sum[i]=sum[i-1]+x;
                        dp[i][i]=0;
                }
                for(len=2; len<=n; len++) {
                        for(i=1; i<=n-len+1; i++) {
                                j=i+len-1;
                                for(k=i+1; k<=j; k++) {
                                        dp[i][j]=min(dp[i][j],dp[i][k-1]+dp[k][j]+sum[j]-sum[i-1]);
                                }
                        }
                }
                printf("%d\n",dp[1][n]);
        }
        return 0;
}

然后这题可以用四边不等式来优化,通过记录s[i][j]的最优分割点为k来将n^3优化成n^2。
优化代码如下:耗时36ms。。。。

#include <iostream>
#include <string.h>
#include <math.h>
#include <queue>
#include <algorithm>
#include <stdlib.h>
#include <map>
#include <set>
#include <stdio.h>
using namespace std;
#define LL long long
#define pi acos(-1.0)
//#pragma comment(linker, "/STACK:1024000000")
const int mod=1e9+7;
const int INF=0x3f3f3f3f;
const double eqs=1e-3;
const int MAXN=40000+10;
int dp[300][300], sum[300], s[300][300];
int main()
{
        int n, i, j, k, len, x;
        while(scanf("%d",&n)!=EOF){
                sum[0]=0;
                memset(dp,INF,sizeof(dp));
                for(i=1;i<=n;i++){
                        scanf("%d",&x);
                        sum[i]=sum[i-1]+x;
                        dp[i][i]=0;
                        s[i][i]=i;
                }
                for(len=2;len<=n;len++){
                        for(i=1;i<=n-len+1;i++){
                                j=i+len-1;
                                for(k=s[i][j-1];k<=s[i+1][j];k++){
                                        if(dp[i][j]>dp[i][k-1]+dp[k][j]+sum[j]-sum[i-1]){
                                                dp[i][j]=dp[i][k-1]+dp[k][j]+sum[j]-sum[i-1];
                                                s[i][j]=k;
                                        }
                                }
                        }
                }
                printf("%d\n",dp[1][n]);
        }
        return 0;
}

NYOJ 737 石子合并(一) (区间DP+平行四边形优化)

标签:acm   算法   编程   dp   

原文地址:http://blog.csdn.net/scf0920/article/details/45250907

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!