哈夫曼树——贪心

哈夫曼树——贪心

哈夫曼树:给定n个权值作为n的叶子结点,构造一棵二叉树,若带权路径长度达到最小,称这样的二叉树为最优二叉树,也称为哈夫曼树(Huffman tree)。哈夫曼树是带权路径长度                 最短的树,权值较大的结点离根较近。

此类题目一般求算哈夫曼树路径总值,利用贪心选择性质每次从队头取出两个边权和最小的树合并成新树入队(优先队列,保证队首元素最小),每次记录新树权值,当队内只剩一棵树时结束算法;

priority_queue<long long,vector<long long>,greater<long long> > q; 



//优先队列,使元素值小的优先权大,(默认元素值大的优先权大);



long long huffman() //哈夫曼树,贪心算法

{

    long long res=0;         //最小代价

    while(q.size()>=2){          //每次合并最小的两个元素,(在队头);

        long long a=q.top();q.pop();

        long long b=q.top();q.pop();

        q.push(a+b);

        res+=a+b;      //计算代价

    }

    return res;

}



int main()

{

    while(cin>>N){

        while(!q.empty()) q.pop();

        while(N--){

            long long tmp;

            cin>>tmp;

            q.push(tmp);    //直接入队

        }

        cout<<huffman()<<endl;

    }

    return 0;

}
huffman贪心算法

 

你可能感兴趣的:(哈夫曼树)