算法训练--重编码

Posted 2021-01-19 drcm

/*
题目：重编码
描述：给每一个字符编一个编码，使得文章的长度达到最短
*/
/*
priority_queue的用法：
https://www.cnblogs.com/Deribs4/p/5657746.html
*/
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
//pq:一个优先级队列，用于辅助后续计算
priority_queue<ll,vector<ll>,greater<ll>> pq;

//w:题目描述中的w（所有）
//n:题目描述中的n
//返回值：答案
long long getAnswer(int n,vector<long,long> w){
//将所有w加入优先队列pq中
for(int i = 0;i < n; ++i){
pq.push(w[i]);
}
ll sum = 0;
while(pq.size() > 1){//当pq中仍有多少个元素时进行循环呢？
ll newEle = 0;//这是本次结束后将要加入到队列的新元素
for(int k = 0;k < 2; ++k) {
newEle = newEle + pq.pop();
pq.pop();
}
sum += newEle; //将本次合并对答案的贡献加入答案
pq.push(newEle); //将新元素加入队列
}
return sum;
}