石子合并问题2——区间DP

石子合并问题2——区间DP_第1张图片

分析:

区间动规一般都是三层for循环,需要注意的是区间用从小到大, 因为动态规划就是后面的用到前面的出的结果递推后面的结果。 dp[i][j] 表示从第 i 堆合并到第 j 堆的最小代价,sum[i][j] 表示第 i 堆到第 j 堆的石子总和。

动态转移方程:dp[i][j] = min(dp[i][j], dp[i][k] + dp[k + 1][j] + sum[i][j]) (i <= k <= j - 1)

import java.util.Scanner;

/**
 * @author Oxygen
 * @date 2018年9月1日
 */
public class Main {
	public static void main(String[] args) {
		Scanner scanner = new Scanner(System.in);
		int N = scanner.nextInt(); //石子有多少堆
		int[] data = new int[N];
		for (int i = 0; i < N; i++) {
			data[i] = scanner.nextInt();
		}
		int[][] dp = new int[N + 1][N + 1]; // dp[i][j] 表示从第 i 堆合并到第 j 堆的最小代价
		for (int i = 0; i < dp.length; i++) {
			for (int j = 0; j < dp[0].length; j++) {
				if (i == j) {
					dp[i][j] = 0;
				} else {
					dp[i][j] = Integer.MAX_VALUE;
				}
			}
		}
		int j = 0;
		for (int d = 1; d < N; d++) { // 前两层用来控制区间长度 
			for (int i = 1; i <= N - d; i++) {
				j = d + i;
				for (int k = i; k < j; k++) { // 最后一层用来枚举最后一次的位置
					dp[i][j] = Math.min(dp[i][j], dp[i][k] + dp[k + 1][j] + sum(data, i-1, j - 1));
				}
			}
		}
		System.out.println(dp[1][N]);
	}

	public static int sum(int[] arr, int i, int j) {
		int res = 0;
		for (int k = i; k <= j; k++) {
			res += arr[k];
		}
		return res;
	}

}

 

你可能感兴趣的:(算法)