見白書
基本思路:
每次肯定選最小的兩個,相加,然後這就是哈夫曼編碼;
代碼如下:
#include<iostream>
#include<iomanip>
#include<algorithm>
#include<queue>
#include<stack>
#include<list>
#include<set>
#include<map>
#include<vector>
#include<cstring>
#include<cstdio>
using namespace std;
typedef long long ll;
const int maxn = 20000+10;
int ans[maxn];
int main()
{
int num;
scanf("%d",&num);
for(int i=0;i<num;i++)
{
scanf("%d",&ans[i]);
}
ll sum=0;
int min1,min2;
while(num>1)
{
if(ans[0]<ans[1])
{
min1=0;
min2=1;
}
else
{
min1=1;
min2=0;
}
for(int i=2;i<num;i++)
{
if(ans[i]<=ans[min1])
{
min2=min1;
min1=i;
}
else if(ans[i]<ans[min2])
{
min2=i;
}
}
int t=ans[min1]+ans[min2];
sum+=t;
if(min1==num-1)
{
ans[min2]=t;
num--;
}
else if(min2==num-1)
{
ans[min1]=t;
num--;
}
else
{
ans[min1]=t;
ans[min2]=ans[num-1];
num--;
}
}
printf("%lld\n",sum);
}