石子归并
Problem Description
N堆石子摆成一条线。现要将石子有次序地合并成一堆。规定每次只能选相邻的2堆石子合并成新的一堆,并将新的一堆石子数记为该次合并的代价。计算将N堆石子合并成一堆的最小代价。
例如: 1 2 3 4,有不少合并方法
1 2 3 4 => 3 3 4(3) => 6 4(9) => 10(19)
1 2 3 4 => 1 5 4(5) => 1 9(14) => 10(24)
1 2 3 4 => 1 2 7(7) => 3 7(10) => 10(20)
括号里面为总代价可以看出,第一种方法的代价最低,现在给出n堆石子的数量,计算最小合并代价。
Input
第1行:N(2 <= N <= 100) 第2 - N + 1:N堆石子的数量(1 <= Ai <= 10000)
Output
输出最小合并代价
Sample Input
4
1
2
3
4
Sample Output
19
解题思路:
区间DP入门题。
合并两个区间的代价是两个区间内石子的权值和。
转移方程:
dp[i][j] = min(dp[i][j],dp[i][k]+dp[k+1][j]+sum[j]-sum[i-1]);
AC代码:
#include <bits/stdc++.h>
#include <iostream>
#include <stdlib.h>
#include <cstdio>
#include <cstring>
#include <utility>
#include <string>
#define INF 0x3f3f3f
#define int long long
using namespace std;
const int mod = 1e9+7;
const int N = 1e2+10;
int a[N];
int dp[N][N];
int sum[N];
signed main()
{
int n;
cin>>n;
for(int i = 1 ; i <= n ; i ++)
{
cin>>a[i];
sum[i] = sum[i-1]+a[i];
}
for(int i = 1 ; i <= n ; i ++)
{
for(int j = 1 ; j <= n ; j ++)
dp[i][j] = INF;
dp[i][i] = 0;
}
for(int len = 0; len < n ; len ++)
{
for(int i = 1 ; i <= n-len ; i ++)
{
int j = i+len;
for(int k = i ; k < j ; k ++)
{
dp[i][j] = min(dp[i][j],dp[i][k]+dp[k+1][j]+sum[j]-sum[i-1]);
}
}
}
cout<<dp[1][n]<<endl;
return 0;
}