学习一发NOIP初赛知识。
/* Pigonometry */
#include <cstdio>
#include <algorithm>
#include <queue>
using namespace std;
typedef long long LL;
int n, k, top;
struct _data {
LL w, h;
bool operator < (const _data &x) const {
return w != x.w ? w > x.w : h > x.h;
}
};
priority_queue<_data> q;
int main() {
scanf("%d%d", &n, &k);
for(int i = 1; i <= n; i++) {
LL w; scanf("%lld", &w);
q.push((_data){w, 0});
}
if((n - 1) % (k - 1) != 0) top = k - 1 - (n - 1) % (k - 1);
for(int i = 1; i <= top; i++)
q.push((_data){0, 0});
top += n;
LL ans = 0;
for(; top != 1; top -= k - 1) {
LL w = 0, h = 0;
for(int i = 1; i <= k; i++) {
_data x = q.top(); q.pop();
w += x.w; h = max(h, x.h);
}
ans += w;
q.push((_data){w, h + 1});
}
printf("%lld\n%lld\n", ans, q.top().h);
return 0;
}