石子合并(一) 区间DP

石子合并（一）(传送门)
时间限制：1000 ms | 内存限制：65535 KB
难度：3
描述
有N堆石子排成一排，每堆石子有一定的数量。现要将N堆石子并成为一堆。合并的过程只能每次将相邻的两堆石子堆成一堆，每次合并花费的代价为这两堆石子的和，经过N-1次合并后成为一堆。求出总的代价最小值。
输入
有多组测试数据，输入到文件结束。
每组测试数据第一行有一个整数n，表示有n堆石子。
接下来的一行有n（0< n <200）个数，分别表示这n堆石子的数目，用空格隔开
输出
输出总代价的最小值，占单独的一行
样例输入
3
1 2 3
7
13 7 8 16 21 4 18
样例输出
9
239

PS: 区间dp, 先找长度为2的最优解，再找长度为3的最优解，直到长度为n.
状态转移方程为: dp[i][j] = min(dp[i][j], dp[i][k]+dp[k+1][j]+a[i]+a[i+1]+…+a[j]);
其中，dp[i][j] 代表区间[i,j]合并所花费的最小代价。
时间复杂度: O(n^3)

#include<iostream>
#include<cstring>
#define Maxn 0x3f3f3f3f
using namespace std;
int dp[1004][1004];
int main()
{
    int n;
    int a[1004];
    int sum[1004];
    while(cin>>n)
    {
        memset(sum, 0, sizeof sum);
        memset(dp, 0, sizeof dp);
        cin>>a[0]; sum[0] = a[0];
        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 = 0; i + len < n; i++)
            {//枚举开始位置
                int j = i + len;
                int temp;
                if(i == 0)temp = sum[j];
                else temp = sum[j] - sum[i-1];
                dp[i][j] = Maxn;
                for(int k = i; k < j; k++)
                {//枚举分割点
                    dp[i][j] = min(dp[i][j], dp[i][k] + dp[k+1][j] + temp);
                }
            }
        }
        cout<<dp[0][n-1]<<endl;
    }
    return 0;
}

这道题可以使用四边形不等式优化，将时间复杂度降低为O(n^2). 设d为区间[i,j]最优分割点，即d = div[i][j]。 那么 $div[i][j-1] <= d <= div[i+1][j]$ .

#include<iostream>
#include<cstring>
#define Maxn 0x3f3f3f3f
using namespace std;
int dp[1004][1004];
int div[1004][1004];
int main()
{
    int n;
    int a[1004];
    int sum[1004];
    while(cin>>n)
    {
        memset(sum, 0, sizeof sum);
        memset(dp, 0, sizeof dp);
        memset(div, 0, sizeof div);
        cin>>a[0]; sum[0] = a[0];
        for(int i = 1; i < n; i++)
        {
            cin>>a[i];
            sum[i] = sum[i-1] + a[i];
        }
        for(int i = 0; i < n - 1; i++)
        {//先计算长度为2的分割点
            dp[i][i+1] = a[i]+a[i+1];
            div[i][i+1] = i;
        }
        for(int len = 2; len < n; len++)
        {//枚举区间长度
            for(int i = 0; i + len < n; i++)
            {//枚举开始位置
                int j = i + len;
                int temp;
                if(i == 0)temp = sum[j];
                else temp = sum[j] - sum[i-1];
                dp[i][j] = Maxn;
                for(int k = div[i][j-1]; k <= div[i+1][j]; k++)
                {//枚举分割点
                    if(dp[i][j] > dp[i][k] + dp[k+1][j] + temp)
                    {
                        div[i][j] = k;//更新分割点
                        dp[i][j] = dp[i][k] + dp[k+1][j] + temp;
                    }
                }
            }
        }
        cout<<dp[0][n-1]<<endl;
    }
    return 0;
}