（CSU - 1592）石子归并

（CSU - 1592）石子归并

Time Limit: 1 Sec Memory Limit: 128 Mb Submitted: 673 Solved: 316

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题。设f[i][j]表示区间[i,j]中所有石子合并所能产生的最小代价，易得状态转移方程为，$f[i][j]=min(f[i][j],f[i][k]+f[k+1][j]+sum[j]-sum[i-1])$ （其中$i<=k<j$，sum数组记录了前缀和,初始值为f[i][i]=0）

递推写法：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

const int INF=0x3f3f3f3f;
const int maxn=105;
int a[maxn],f[maxn][maxn],sum[maxn];

int main()
{
    int T,n;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        sum[0]=0;
        for(int i=1;i<=n;i++) 
        {
            scanf("%d",a+i);
            sum[i]=sum[i-1]+a[i];
            f[i][i]=0;
        }
        for(int l=2;l<=n;l++)//枚举长度 
            for(int i=1;i+l-1<=n;i++)//枚举起点 
            {
                int L=i,R=i+l-1;
                f[L][R]=INF;
                for(int k=i;k<i+l-1;k++)//枚举中间点 
                    f[L][R]=min(f[L][R],f[L][k]+f[k+1][R]+sum[R]-sum[L-1]);
            }
        printf("%d\n",f[1][n]);
    }
    return 0;
}

记忆化搜索写法：

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;

const int INF=0x3f3f3f3f;
const int maxn=105;
int f[maxn][maxn];
int a[maxn];
int pre[maxn];

int dfs(int i,int j)
{
    if(f[i][j]!=-1) return f[i][j];
    if(i==j) return 0;//递归结束条件 
    int ans=INF;
    for(int k=i;k<j;k++)
        ans=min(ans,dfs(i,k)+dfs(k+1,j)+pre[j]-pre[i-1]);
    f[i][j]=ans;
    return ans;
}

int main()
{
    int T,n;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        pre[0]=0;
        for(int i=1;i<=n;i++)
        {
            scanf("%d",a+i);
            pre[i]=pre[i-1]+a[i];
        }
        memset(f,-1,sizeof(f));
        printf("%d\n",dfs(1,n)); 
    }
    return 0;
}