1. 程式人生 > >[其它-GarsiaWachs演算法]51nod 1023 石子歸併v3

[其它-GarsiaWachs演算法]51nod 1023 石子歸併v3

基準時間限制:2 秒 空間限制:131072 KB 分值: 640 難度:8級演算法題

N堆石子擺成一條線。現要將石子有次序地合併成一堆。規定每次只能選相鄰的2堆石子合併成新的一堆,並將新的一堆石子數記為該次合併的代價。計算將N堆石子合併成一堆的最小代價。

例如: 1 2 3 4,有不少合併方法

1 2 3 4 => 3 3 4(3) => 6 4(9) => 10(19)

1 2 3 4 => 1 5 4(5) => 1 9(14) => 10(24)

1 2 3 4 => 1 2 7(7) => 3 7(10) => 10(20)

括號裡面為總代價可以看出,第一種方法的代價最低,現在給出n堆石子的數量,計算最小合併代價。

Input

第1行:N(2 <= N <= 50000)

第2 - N + 1:N堆石子的數量(1 <= A[i] <= 10000)

Output

輸出最小合併代價

Input示例

4

1

2

3

4

Output示例

19

這道題用GarsiaWachs演算法,時間複雜度達到O(nlogn)操作過程如下:

對於剩下來的k堆石子,如果存在一個最小的i,滿足data[i-2]<=data[i],那麼就可以優先合併data[i-2]和data[i-1]這兩堆。

證明略。。。。。。。

拿程式碼感受一下。

#include <stdio.h>
#include <algorithm>
#include <iostream>
#include <string.h>
#define MAXN 50005
#define LL long long
using namespace std;
int n, num;
LL ans;
int data[MAXN];
void dfs(int now) { int j; int temp = data[now - 1] + data[now];//代價 ans += (LL)temp; for(int i = now; i < num - 1; i++) data[i] = data[i + 1]; num--; for(j = now - 1; j > 0 && data[j - 1] < temp; j--) data[j] = data[j - 1]; data[j] = temp; while(j >= 2 && data[j - 2] <= data[j]) { int d = num - j; dfs(j - 1); j = num - d; } } int main() { scanf("%d", &n); for(int i = 0; i < n; i++) scanf("%d", &data[i]); num = 1; ans = 0; for(int i = 1; i < n; i++) { //printf("%d %d\n",num,i); data[num++] = data[i]; while(num>=3 && data[num-3]<=data[num-1]) dfs(num - 2); } while(num > 1) dfs(num - 1); printf("%lld\n", ans); return 0; }