石子合并(区间DP)
1.题目
2.基本思想
状态计算:
(1) i<j 时,f[i][j] = min ( f[i][k] + f[k+1][j] + s[j] − s[i−1] ) ( i ≤ k ≤ j−1)
(2)i=j 时, f[i][i]=0 (合并一堆石子代价为 0)
3.代码实现
时间复杂度O(n^3)
import java.util.Scanner;
public class Main{
static final int N=310;
public static void main(String args[]){
Scanner sc=new Scanner(System.in);
int n=sc.nextInt();
int s[] =new int[N+1];
int f[][]=new int[N][N];
for(int i=1;i<=n;i++){
s[i]=sc.nextInt();
s[i]+=s[i-1];
}
//枚举区间长度
for(int len= 2;len<=n;len++){
//枚举起点i
for(int i=1;i+len-1<=n;i++){
int left = i , right = i+len-1;
f[left][right]=Integer.MAX_VALUE;
//枚举划分位置k 取不到最后一个 需要留一个
for(int k=left;k<right;k++){
f[left][right]=Math.min(f[left][right],f[left][k]+f[k+1][right]+s[right]-s[left-1]);
//s[right] - s[left - 1]):表示合并f[left][right], f[left][k] + f[k + 1][right]的代价
}
}
}
System.out.println(f[1][n]);
}
}
4.总结(区间DP常用模板)
区间DP问题枚举时,第一维通常时枚举区间长度,并且一般 len = 1 时用来初始化, 枚举从 len = 2 开始;第二维枚举起点 i (右端点 j : j= i + len -1 )
代码模板如下:
for (int len = 1; len <= n; len++) { // 区间长度
for (int i = 1; i + len - 1 <= n; i++) { // 枚举起点
int j = i + len - 1; // 区间终点
if (len == 1) {
dp[i][j] = 初始值
continue;
}
for (int k = i; k < j; k++) { // 枚举分割点,构造状态转移方程
dp[i][j] = min(dp[i][j], dp[i][k] + dp[k + 1][j] + w[i][j]);
}
}
}