最优合并问题

最新推荐文章于 2020-07-01 16:17:09 发布

南极墨白

最新推荐文章于 2020-07-01 16:17:09 发布

阅读量835

点赞数

文章标签：算法队列数据结构贪心算法 huffman tree

本文链接：https://blog.csdn.net/qq_43542311/article/details/105924392

版权

问题描述

给定k个排好序的序列s1,s2, …, sk，用2路合并算法将这k个序列合并成一个序列。假设采用的2路合并算法合并2个长度分别为m和n的序列需要m+n-1次比较。试设计一个算法确定合并这个序列的最优合并顺序，使所需的总比较次数最少。

为了进行比较，还需要确定合并这个序列的最差合并顺序，使所需的总比较次数最多。

算法分析

对于给定的k个待合并序列，计算最多比较次数和最少比较次数合并方案，可采用huffman算法，贪心策略：每次选最短的序列合并得到最少比较次数；每次选最长的序列合并得到最多比较次数，这里使用优先队列来进行数据的保存

代码

#include <iostream>
#include <queue>
using namespace std;
priority_queue<int, vector<int>, greater<int> > greater_queue;
priority_queue<int, vector<int>, less<int> > less_queue;

template <typename T>
int solve(T q)
{
    int ans = 0;
    int sum = 0;
    while (q.size() > 1)
    {
        sum = 0;
        for (int i = 0; i < 2; i++)
        {
            sum += q.top();
            q.pop();
        }
        ans += (sum-1);
        q.push(sum);
    }
    return ans;
}

int main()
{
    int n;
    cin >> n;
    for (int i = 0, temp; i < n; i++)
    {
        cin >> temp;
        greater_queue.push(temp);
        less_queue.push(temp);
    }
    cout << solve(less_queue) << " " << solve(greater_queue) << endl;
}