. 荷馬史詩
追逐影子的人,自己就是影子。 ——荷馬
達達最近迷上了文學。
她喜歡在一個慵懶的午後,細細地品上一杯卡布奇諾,靜靜地閱讀她愛不釋手的《荷馬史詩》。
但是由《奧德賽》和《伊利亞特》組成的鴻篇鉅製《荷馬史詩》實在是太長了,達達想通過一種編碼方式使得它變得短一些。
一部《荷馬史詩》中有 n 種不同的單詞,從 1 到 n 進行編號。其中第 i 種單詞出現的總次數為 wi<?XML:NAMESPACE PREFIX = "[default] http://www.w3.org/1998/Math/MathML" NS = "http://www.w3.org/1998/Math/MathML" />wi。
達達想要用 k 進位制串 sisi 來替換第 i 種單詞,使得其滿足如下要求:
對於任意的 1≤i,j≤n,i≠j,都有:sisi 不是 sjsj 的字首。
現在達達想要知道,如何選擇 sisi,才能使替換以後得到的新的《荷馬史詩》長度最小。
在確保總長度最小的情況下,達達還想知道最長的 sisi 的最短長度是多少?
一個字串被稱為 k 進位制字串,當且僅當它的每個字元是 0 到 k−1 之間(包括 0 和 k−1)的整數。
字串 Str1 被稱為字串 Str2 的字首,當且僅當:存在 1≤t≤m,使得 Str1=Str2[1..t]。
其中,m 是字串 Str2 的長度,Str2[1..t] 表示 Str2 的前 t 個字元組成的字串。
注意:請使用64位整數進行輸入輸出、儲存和計算。
輸入格式
輸入檔案的第 1 行包含 2 個正整數 n,k,中間用單個空格隔開,表示共有 n 種單詞,需要使用 k 進位制字串進行替換。
第2~n+1 行:第 i+1 行包含 1 個非負整數 wiwi,表示第 i 種單詞的出現次數。
輸出格式
輸出檔案包括 2 行。
第 1 行輸出 1 個整數,為《荷馬史詩》經過重新編碼以後的最短長度。
第 2 行輸出 1 個整數,為保證最短總長度的情況下,最長字串 sisi 的最短長度。
資料範圍
2≤n≤1000002≤n≤100000,
2≤k≤92≤k≤9
1≤wi≤10121≤wi≤1012
輸入樣例:
4 2
1
1
2
2
輸出樣例:
12
2
#include <iostream> #include <cstdio> #include <queue> #include <vector> using namespace std; typedef long long LL; typedef pair<LL,int> PLI; int n,k; priority_queue<PLI,vector<PLI> ,greater<PLI>> heap; int main(){ scanf("%d%d",&n,&k); for (int i = 0; i < n; i++){ LL x; scanf("%lld",&x); heap.push({x,0}); } while ((n - 1) % (k - 1) != 0)heap.push({0,0}),n++; LL ans = 0; while(heap.size() > 1){ int h = 0; LL d = 0; for (int i = 0; i < k; i++){ d += heap.top().first; h = max(h,heap.top().second + 1); heap.pop(); } heap.push({d,h}); ans += d; } cout << ans << endl << heap.top().second; return 0; }