POJ1651-Multiplication Puzzle-区间DP

题目: http://poj.org/problem?id=1651

本来想自己写, 虽然有点大体思路, 但是一直没有最终定型, 最后还是参考别人的思路和代码才算明白了.

刚开始找了好几篇博客, 发现虽然也都是用区间DP写的, 但是并不是完全基于这道题的基本思想写的. 很多都是将这道题抽象为矩阵连乘, 然后运用区间DP求解, 输出结果dp[1][n - 1].

最后看了一篇博客才最后明白了直接基于这道题的区间DP求解方式.

内容链接: http://blog.sina.com.cn/s/blog_4cb72c560100kwym.html

题目要求是在一个数字序列中, 取出不包括头和尾的所有数字, 每次取出一个数字的代价等于取出的这个数和左右两个数的乘积, 求总代价和最小的取法.

由题目要求可以知道, 最后序列中一定只剩头和尾两个数. 因此我们可以反过来求解. 即:对头尾两个数之间插入数字还原成原序列的方法进行求解, 每次插入时遍历插入位置, 从最小的区间段开始插入, 然后递增扩展至全部区间即可得结果.

dp[i][j] 表示在把区间[i...j]内的所有数字插入之后的最小代价; 其实也可以这样理解: dp[i][j]表示取出区间[i...j]之间的所有数字的最小代价.

初始化:

if(i = j - 1)

   dp[i][j] = 0;

else

   dp[i][j] = INF;

区间[i...j]的最小长度为1, 因为长度为0的则无法在其中插入数字(可就是区间长度为0的话就不满足题意了, 无法计算从中取出数字的代价dp[i][j], 当区间长度为1时, 取出其中数字的代价d[i][j]=0, 因为num[i]和num[j]之间没有数字了)

遍历区间的起点i和长度len, 假设在区间[i...j]之间的k位置插入数字, 则dp[i][j] = dp[i][k] + dp[k][j] + num[i] * num[k] * num[j];

因为是从最小的区间段len = 1;开始计算的, 所以后面递推计算dp[i][j]时所需要的dp[i][k]和dp[k][j]已经计算出来了(因为[i...k]的区间长度和[k...j]的区间长度小于[i...j]的区间长度).

最后输出dp[1][n]即为最终结果!

#include <stdio.h>
#include <string.h>
#include <math.h>
#include <iostream>
#include <algorithm>
#include <string>
#include <queue>
#include <stack>
#include <vector>
#include <list>
#include <map>
#include <set>
#include <iomanip>
#define CL(x,v); memset(x,v,sizeof(x));
#define INF 1<<29
#pragma comment(linker, "/STACK:1024000000,1024000000")
using namespace std;

const int N = 100 + 10;
int num[N], dp[N][N];

int main()
{
	int n;
	 while(cin >> n)
     {
         CL(dp, 0);
         for(int i = 1; i <= n; ++i)
            cin >> num[i];

         for(int len = 2; len <= n - 1; ++len) //区间长度由2开始, 省去dp[i][j] = 0;的赋值计算, 上面已经完成
         {
             for(int i = 1; i <= n - len; ++i)
             {
                 int j = i + len;
                 dp[i][j] = INF;
                 for(int k = i + 1; k <= j - 1; ++k) //k = i + 1;和k <= j - 1;也是因为区间的最小长度为1
                     dp[i][j] = min(dp[i][j], dp[i][k] + dp[k][j] + num[i] * num[k] * num[j]);
             }
         }
         cout << dp[1][n] << endl;
     }
     return 0;
}