poj 3253

来源:互联网 发布:c语言设置线程优先级 编辑:程序博客网 时间:2024/05/21 13:59
题目大意:

见白书

基本思路:
每次肯定选最小的两个,相加,然后这就是哈夫曼编码;

代码如下:

#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);
}

原创粉丝点击