蓝桥杯 试题 基础练习 Huffuman树

#include<bits/stdc++.h>

using namespace std;

multiset<int> mp;//允许有重复数字并且从小到大排列 
int main(){
	int n;
	cin>>n;
	for(int i=0;i<n;i++){
		int res;
		cin>>res;
		mp.insert(res);
	}
	int ans = 0;//计算总数
	int flag;
	for(int i=1;i<=n-1;i++){//得到一个数需要执行n-1次操作 
	   flag = 0;
	   flag = *mp.begin();//加第一个数 
	   mp.erase(mp.begin());//删除 
	   flag += *mp.begin(); //加第二个数
	   mp.erase(mp.begin());//删除
	   mp.insert(flag);
	   ans += flag; 
	} 
	cout<<ans<<endl;
	return 0;
}