最优二叉树c++代码——优先队列

最优二叉树
传统题 1000ms 128MiB

Description
现在有N个正整数，每一次去掉其中2个数a和b，然后加入一个数a*b+1，这样最后只剩下一个数p。 要求求出最大的p记为maxp，最小的p记为minp，和他们的差K=maxp-minp。

对于给定的数列，编程计算出它的max,min和K。

Input Format
第一行是数列的长度N（不超过50），第二行共n个正整数（不超过30）。

Output Format
输出一共三行，每行一个整数，依次为max,min,K。

Sample
样例输入
2
1 1
样例输出
2
2
0

---

思路：首先定义两个优先队列 max,min,K变量 按题目要求去掉其中2个数a和b，然后加入一个数a*b+1 优先队列队首即为最大值 最后用变量K计算差值并输出 注意数据要定义long long

---

代码：

#include<iostream>
#include<queue>
using namespace std;
typedef long long ll;

int a[55];

priority_queue<ll,vector<ll>,greater<ll> >q1;
priority_queue<ll,vector<ll>,less<ll> >q2;

int main(){
    int n;
    cin>>n;

    for(int i=0;i<n;i++){
        cin>>a[i];
        q1.push(a[i]);
        q2.push(a[i]);
    }

    ll x,y;
    ll maxp,minp,K;

    while(q1.size()>1){
        x=q1.top(); q1.pop();
        y=q1.top(); q1.pop();
        q1.push(x*y+1);
    }
    maxp=q1.top();

    while(q2.size()>1){
        x=q2.top(); q2.pop();
        y=q2.top(); q2.pop();
        q2.push(x*y+1);
    }
    minp=q2.top();

    K=maxp-minp;
    cout<<maxp<<endl<<minp<<endl<<K;
    return 0;
}