[其它-GarsiaWachs演算法]51nod 1023 石子歸併v3
阿新 • • 發佈:2019-02-04
基準時間限制: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; }