1. 程式人生 > 實用技巧 >. 荷馬史詩

. 荷馬史詩

追逐影子的人,自己就是影子。 ——荷馬

達達最近迷上了文學。

她喜歡在一個慵懶的午後,細細地品上一杯卡布奇諾,靜靜地閱讀她愛不釋手的《荷馬史詩》。

但是由《奧德賽》和《伊利亞特》組成的鴻篇鉅製《荷馬史詩》實在是太長了,達達想通過一種編碼方式使得它變得短一些。

一部《荷馬史詩》中有 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;
}