最优合并问题 SDUT OJ 贪心算法

最优合并问题

Description

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

Input

输入数据的第一行有1 个正整数k（k≤1000），表示有k个待合并序列。接下来的1 行中，有k个正整数，表示k个待合并序列的长度。

Output

输出两个整数，中间用空格隔开，表示计算出的最多比较次数和最少比较次数。

Sample

Input 

4
5 12 11 2

Output 

78 52

 

解题思路：

/***
每次合并两个序列时，令总比较次数增加 m+n-1

确定合并这个序列的最优合并顺序，使所需的总比较次数最少：
要想使总比较次数最少，需要先合并短的序列，
使短序列合并最多次，使长序列合并最少次

确定合并这个序列的最差合并顺序，使所需的总比较次数最多：
要想使总比较次数最多，需要先合并长的序列，
使长序列合并最多次，使短序列合并最少次

优先队列知识 http://c.biancheng.net/view/6987.html
***/

参考同类型题：多元Huffman编码问题 SDUT OJ 贪心算法 

https://blog.csdn.net/weixin_44143702/article/details/110245358

#include <iostream>
#include <queue> ///优先队列
using namespace std;
/***
每次合并两个序列时，令总比较次数增加 m+n-1

确定合并这个序列的最优合并顺序，使所需的总比较次数最少：
要想使总比较次数最少，需要先合并短的序列，
使短序列合并最多次，使长序列合并最少次

确定合并这个序列的最差合并顺序，使所需的总比较次数最多：
要想使总比较次数最多，需要先合并长的序列，
使长序列合并最多次，使短序列合并最少次

优先队列知识 http://c.biancheng.net/view/6987.html
***/

priority_queue <int, vector<int>, less<int> > q1;///降序队列
priority_queue <int, vector<int>, greater<int> > q2;///升序队列
void get_maxsum();///计算最多比较次数
void get_minsum();///计算最少比较次数
int max_sum;
int min_sum;
int main()
{
    int k;
    int x;///临时变量，用来转移读入的数据
    cin >> k;
    for(int i = 0; i < k; i++)
    {
        cin >> x;
        q1.push(x);
        q2.push(x);
    }
    max_sum = min_sum = 0;
    get_maxsum();
    get_minsum();
    cout << max_sum << " " << min_sum << endl;
    return 0;
}
void get_maxsum()
{
    int sum;///临时变量，用来计算两序列合并后的长度
    while(q1.size() > 1)
    {
        sum = q1.top();///sum保存队列顶的两个最大的数相加的值
        q1.pop();
        sum += q1.top();
        q1.pop();
        q1.push(sum);///每次计算完合并的值记得放回去队列
        max_sum += sum - 1;///比较次数等于m+n-1 即 两序列和-1(sum-1)
    }
}
void get_minsum()
{
    int sum;///临时变量，用来计算两序列合并后的长度
    while(q2.size() > 1)
    {
        sum = q2.top();///sum保存队列顶的两个最小的数相加的值
        q2.pop();
        sum += q2.top();
        q2.pop();
        q2.push(sum);///每次计算完合并的值记得放回去队列
        min_sum += sum - 1;///比较次数等于m+n-1 即 两序列和-1(sum-1)
    }
}