(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;
}