码迷,mamicode.com
首页 > 其他好文 > 详细

UVA 10954 Add All 全部相加 (Huffman编码)

时间:2015-08-09 18:25:37      阅读:103      评论:0      收藏:0      [点我收藏+]

标签:

题意:给你n个数的集合,每次选两个删除,把它们的和放回集合,直到集合的数只剩下一个,每次操作的开销是那两个数的和,求最小开销。

赤果果的Huffman码了。用两个队列类似归并排序,合并一下。

#include<bits/stdc++.h>
using namespace std;
const int maxn = 5010;
int q1[maxn];
int q2[maxn];

#define GetMin(x)if(head1>=rear1 || (head2<rear2 && q1[head1]>q2[head2])) x = q2[head2++];else x = q1[head1++];

int main()
{
    int n,x;
    while(scanf("%d",&n),n){
        int head1 = 0,rear1 = n,head2 = 0,rear2 = 0;
        for(int i = 0; i < n; i++) scanf("%d",q1+i);
        int ans = 0;

        sort(q1,q1+n);
        int a = q1[head1++];
        int b = q1[head1++];
        ans += q2[rear2++] = a+b;
        for(int i = 2; i < n; i++){
            GetMin(a) GetMin(b)
            ans += q2[rear2++] = a+b;
        }
        printf("%d\n",ans);
    }
    return 0;
}

效率要求不高的时候可以用优先队列

#include<bits/stdc++.h>
using namespace std;

int main()
{
    int n,x;
    while(~scanf("%d",&n)&&n){
        priority_queue<int,vector<int>,greater<int> >q;
        for(int i = 0; i < n ;i++) { scanf("%d",&x); q.push(x); }
        int ans = 0;
        for(int i = 1; i < n; i++){
            int a = q.top(); q.pop();
            int b = q.top(); q.pop();
            ans += a+b;
            q.push(a+b);
        }
        printf("%d\n",ans);
    }
    return 0;
}

 

UVA 10954 Add All 全部相加 (Huffman编码)

标签:

原文地址:http://www.cnblogs.com/jerryRey/p/4715480.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!