藍橋杯 石子合併

問題描述
  在一條直線上有n堆石子,每堆有一定的數量,每次可以將兩堆相鄰的石子合併,合併後放在兩堆的中間位置,合併的費用爲兩堆石子的總數。求把所有石子合併成一堆的最小花費。
輸入格式
  輸入第一行包含一個整數n,表示石子的堆數。
  接下來一行,包含n個整數,按順序給出每堆石子的大小 。
輸出格式
  輸出一個整數,表示合併的最小花費。
樣例輸入
5
1 2 3 4 5
樣例輸出
33
數據規模和約定
  1<=n<=1000, 每堆石子至少1顆,最多10000顆。
只得了90分,一組數據超時。
#include<iostream>
#include<cstdio>
#include<algorithm>
#define MAX 1005
using namespace std;
int arr[MAX], Min[MAX][MAX],s[MAX];
void dp(int n)
{
	int i, j,k, r, t;
	for (int i = 1;i <= n;i++)
		Min[i][i] = 0;
	for (r = 2;r <= n;r++)
		for (i = 1;i <= n-r+1;i++)
		{
			j = i + r - 1;
			Min[i][j] = Min[i + 1][j] +s[j]-s[i-1];
			for (k = i+1;k < j;k++)
				t = Min[i][k] + Min[k + 1][j]+s[j]-s[i-1],
				Min[i][j] = min(Min[i][j], t);
		}
}
int main()
{
	int n;
	cin >> n;
	for (int i = 1;i <= n;i++)
		scanf("%d", &arr[i]),s[i]=s[i-1]+arr[i];
	dp(n);
	cout << Min[1][n];
	return 0;
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章