1760多元Huffman編碼問題(貪心)
阿新 • • 發佈:2020-12-06
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; 22q1.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 }