石子合并:
链接:登录—专业IT笔试面试备考平台_牛客网
来源:牛客网
设有N堆沙子排成一排,其编号为1,2,3,…,N1,2,3,\dots ,N1,2,3,…,N(N≤300)(N\leq 300)(N≤300)。每堆沙子有一定的数量,可以用一个整数来描述,现在要将这N堆沙子合并成为一堆,每次只能合并相邻的两堆,合并的代价为这两堆沙子的数量之和,合并后与这两堆沙子相邻的沙子将和新堆相邻,合并时由于选择的顺序不同,合并的总代价也不相同,如有4堆沙子分别为 1 3 5 2 我们可以先合并1、2堆,代价为4,得到4 5 2 又合并 1,2堆,代价为9,得到9 2 ,再合并得到11,总代价为4+9+11=24,如果第二步是先合并2,3堆,则代价为7,得到4 7,最后一次合并代价为11,总代价为4+7+11=22;问题是:找出一种合理的方法,使总的代价最小。输出最小代价。
分析:
对于我们的一个区间,假设我们在之前已经维护了这个区间内的一个个小区间(区间dp就是从小区间到大区间的,所以是可以维护的),那么我们显然的就可以使用小区间来递推到大区间,因为显然的一个大区间可以分成两个小区间,也就是说我们使用一个中间值作为我们的大区间的分割点的话,我们可以得到以下的转移方程。
dp[i][j] = Math.min(dp[i][j], dp[i][k] + dp[k + 1][j] + sum[j] - sum[i - 1]);
完整代码:
import java.util.Arrays;
import java.util.Scanner;
public class Main {
static int max = 0x3f3f3f3f;
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
while (sc.hasNext()) {
int n = sc.nextInt();
int[] num = new int[n + 1];
int[] sum = new int[n + 1];
int[][] dp = new int[n + 1][n + 1];
for (int i = 0; i <= n; i++) {
Arrays.fill(dp[i], max);
}
sum[0] = 0;
for (int i = 1; i <= n; i++) {
num[i] = sc.nextInt();
sum[i] = num[i] + sum[i - 1];
dp[i][i] = 0;
}
for (int len = 1; len < n; len++) {
for (int i = 1; i + len <= n; i++) {
int j = i + len;
for (int k = i; k < j; k++) {
dp[i][j] = Math.min(dp[i][j], dp[i][k] + dp[k + 1][j] + sum[j] - sum[i - 1]);
}
}
}
System.out.println(dp[1][n]);
}
}
}