1. 程式人生 > 實用技巧 >1760多元Huffman編碼問題(貪心)

1760多元Huffman編碼問題(貪心)

Description

在一個操場的四周擺放著n堆石子。現要將石子有次序地合併成一堆。規定每次至少選2 堆最多選k堆石子合併成新的一堆,合併的費用為新的一堆的石子數。試設計一個演算法,計算出將n堆石子合併成一堆的最大總費用和最小總費用。
對於給定n堆石子,計算合併成一堆的最大總費用和最小總費用。

Input

輸入資料的第1 行有2 個正整數n和k(n≤100000,k≤10000),表示有n堆石子,每次至少選2 堆最多選k堆石子合併。第2 行有n個數(每個數均不超過 100),分別表示每堆石子的個數。

Output

將計算出的最大總費用和最小總費用輸出,兩個整數之間用空格分開。

Sample

Input

7 3
45 13 12 16 9 5 22

Output

593 199

Hint

請注意資料範圍是否可能爆 int。

 1 #include <iostream>
 2 #include <algorithm>
 3 #include <string.h>
 4 #include <vector>
 5 #include <queue>
 6 
 7 #define inf 0x3f3f3f3f
 8 
 9 using namespace std;
10 
11 priority_queue<int, vector<int
>, greater<int> > q1;///greater是從小到大 12 priority_queue<int, vector<int>, less<int> >q2;///less是從大到小 13 14 int main() 15 { 16 int n, k, i, x; 17 long long sum1, sum2, sum; 18 cin >> n >> k; 19 for(i=0;i<n;i++) 20 { 21 cin >> x; 22
q1.push(x); 23 q2.push(x); 24 } 25 while(q1.size()%(k-1) != 1) 26 { 27 q1.push(0); 28 } 29 sum1 = 0; 30 while(q1.size()!=1) 31 { 32 sum = 0; 33 for(i=0;i<k;i++) 34 { 35 sum += q1.top(); 36 q1.pop(); 37 } 38 sum1 += sum; 39 q1.push(sum); 40 } 41 sum2 = 0; 42 while(q2.size()!=1) 43 { 44 sum = 0; 45 for(i=0;i<2;i++) 46 { 47 sum += q2.top(); 48 q2.pop(); 49 } 50 sum2 += sum; 51 q2.push(sum); 52 } 53 cout << sum2 << " " << sum1 << endl; 54 return 0; 55 }