Java实现 蓝桥杯 算法提高 合并石子

算法提高 合并石子

时间限制：2.0s 内存限制：256.0MB

问题描述

　　在一条直线上有n堆石子，每堆有一定的数量，每次可以将两堆相邻的石子合并，合并后放在两堆的中间位置，合并的费用为两堆石子的总数。求把所有石子合并成一堆的最小花费。

输入格式

　　输入第一行包含一个整数n，表示石子的堆数。

　　接下来一行，包含n个整数，按顺序给出每堆石子的大小 。

输出格式

　　输出一个整数，表示合并的最小花费。

样例输入

5

1 2 3 4 5

样例输出

33

数据规模和约定

　　1<=n<=1000, 每堆石子至少1颗，最多10000颗。

import java.util.Scanner;

public class 合并石子 {
	 public static void main(String[] args) {
         Scanner in = new Scanner(System.in);
         int n = in.nextInt();
         int[] a = new int[1010];
         int[][] temp = new int[1010][1010];
         int[][] dp = new int[1010][1010];
         for (int i=1; i<=n; i++) {
             a[i] = in.nextInt();
             temp[i][i] = a[i];
         }
         for (int i=1; i<n; i++) {
             for(int j=i+1; j<=n; j++) {
                 temp[i][j] = temp[i][j-1] + a[j];
             }
         }
         for (int r=2; r<=n; r++) {
             for (int i=1; i<=n-r+1; i++) {
                 int j=i+r-1;
                 dp[i][j] = Integer.MAX_VALUE;
                 for (int k=i; k<j; k++) {
                     if(dp[i][j] > dp[i][k] + dp[k+1][j])
                         dp[i][j] = dp[i][k] + dp[k+1][j];
                 }
                 dp[i][j] += temp[i][j];
             }
         }
         System.out.println(dp[1][n]);
     }  

}