現在有n堆石子,第i堆有ai個石子。現在要把這些石子合併成一堆,每次只能合併相鄰兩個,每次合併的代價是兩堆石子的總石子數。求合併所有石子的最小代價。
Input第一行包含一個整數T(T<=50),表示數據組數。
每組數據第一行包含一個整數n(2<=n<=100),表示石子的堆數。
第二行包含n個正整數ai(ai<=100),表示每堆石子的石子數。
每組數據僅一行,表示最小合併代價。
Sample Input2 4 1 2 3 4 5 3 5 2 1 4Sample Output
19 33
區間dp,狀態轉移方程
dp[i][i+l-1]=min(dp[i][i+l-1],dp[i][k]+dp[k+1][i+l-1]+sum[i+l-1]-sum[i-1]);
感覺區間dp全難在寫這個分治上了
#include<iostream>
#include<cstring>
using namespace std;
int n,a[105],dp[105][105],sum[105];
int main()
{
int T;
cin>>T;
while(T--)
{
memset(dp,0,sizeof(dp));
memset(sum,0,sizeof(sum));
cin>>n;
for(int i=1;i<=n;i++)
{
cin>>a[i];
sum[i]=sum[i-1]+a[i];
}
for(int l=2;l<=n;l++)
for(int i=1;i<=n-l+1;i++)
{
dp[i][i+l-1]=1000000;
for(int k=i;k<=i+l-2;k++)
dp[i][i+l-1]=min(dp[i][i+l-1],dp[i][k]+dp[k+1][i+l-1]+sum[i+l-1]-sum[i-1]);
}
cout<<dp[1][n]<<endl;
}
return 0;
}