CSU 1592：石子归并（区间DP）

石子归并

Time limit:20000 ms Memory limit:131072 kB

---

Problem Description

    现在有n堆石子，第i堆有ai个石子。现在要把这些石子合并成一堆，每次只能合并相邻两个，每次合并的代价是两堆石子的总石子数。求合并所有石子的最小代价。

Input

第一行包含一个整数T（T<=50），表示数据组数。
每组数据第一行包含一个整数n（2<=n<=100），表示石子的堆数。
第二行包含n个正整数ai（ai<=100），表示每堆石子的石子数。

Output

每组数据仅一行，表示最小合并代价。

Sample Input

2
4
1 2 3 4
5
3 5 2 1 4

Sample Output

19
33

---

解题思路：

经典的区间DP，设dp[i][j]是从i~j个石子合并的最小价值，很容易得出

dp[i][j]=min(dp[i][j],dp[i][k]+dp[k+1][j]+c);

$$dp[i][j]=min(dp[i][j],dp[i][k]+dp[k+1][j]+c);$$ 其中c=i~j个石子的总价值。
另一种方法是递归+记忆化查询

int dfs(int i,int j){
   if (dp[i][j]!=-1) 
       return dp[i][j];
   int ans=1e9;
   for (int k=i;k<j;k++) 
       ans=min(ans,dfs(i,k)+dfs(k+1,j)+c[j]-c[i-1]);//c数组代表前i个石子的价值总和
   return dp[i][j]=ans;
}

---

Code：

#include <iostream>
#include <cstring>
#include <algorithm>
#include <cstdio>
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;

const int maxn=100+5;
const int INF=1e9;
int a[maxn];
int dp[maxn][maxn];
int c[maxn];

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int n;
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
        {
            scanf("%d",a+i);
        }
        for(int i=1;i<=n;i++)
        {
            c[i]=c[i-1]+a[i];
        }
        for(int len=2;len<=n;len++)
        {
            for(int i=1;i<=n-len+1;i++)
            {
                int cost=c[i+len-1]-c[i-1];
                dp[i][i+len-1]=INF;
                for(int j=i;j<=i+len-2;j++)
                {
                    dp[i][i+len-1]=min(dp[i][i+len-1],dp[i][j]+dp[j+1][i+len-1]+cost);
                }
            }
        }
        printf("%d\n",dp[1][n]);
    }
    return 0;
}