UESTC 1599 wtmsb 优先队列(最小值优先)

wtmsb

Time Limit: 1000/100MS (Java/Others)     Memory Limit: 131072/131072KB (Java/Others)


Submit  Status
这天，AutSky_JadeK看到了n张图片,他忍不住说道：“我TM社保！”。


每张图片有一个社保值，他可以合并两张图片，合并所得的图片的社保值是原来两张图片的社保值之和。
每次合并需要消耗的体力也是原来两张图片的社保值之和。 显然，n−1次合并之后，只剩下一张图片。
他想知道，在这个过程中，他消耗的总体力最少是多少。

Input

第一行包含一个整数n，
第二行包含n个整数A1,A2,…,An，代表n张图片的社保值。

Output

输出一行，包含一个整数，代表他消耗的总体力最少是多少。

Sample input and output

Sample Input Sample Output
3
1 2 9
15

Hint

1≤n≤200001≤n≤20000，
1≤Ai≤20001≤Ai≤2000

Source

17暑假前集训-数据结构专题 By AutSky_JadeK

2017 UESTC Training for Data Structures

UESTC 1599 wtmsb

My Solution

题意：有n个数，每次可以选择2个数x和y进行合并，其代价和结果都是x+y，要求n-1次合并的最小代价。


优先队列(最小值优先)
把所有的数丢进优先队列，
然后每次如果队列里的元素个数>= 2，
则取出2个数合并记录然后把和丢回队列，
直到元素个数为1.
复杂度 O(nlogn)

#include <iostream>
#include <cstdio>
#include <queue>
using namespace std;
typedef long long LL;
const int MAXN = 2e4 + 8;

priority_queue<LL, vector<LL>, greater<LL>> pq;

int main()
{
    #ifdef LOCAL
    freopen("j.txt", "r", stdin);
    //freopen("j.out", "w", stdout);
    int T = 4;
    while(T--){
    #endif // LOCAL
    //ios::sync_with_stdio(false); cin.tie(0);

    int n, a, i, b;
    LL ans = 0;
    scanf("%d", &n);
    for(i = 0; i < n; i++){
        scanf("%d", &a);
        pq.push(a);
    }
    int sz = pq.size();
    while(sz >= 2){
        a = pq.top();
        pq.pop();
        b = pq.top();
        pq.pop();
        a = a + b;
        ans += a;
        pq.push(a);
        sz--;
    }

    cout << ans << endl;

    #ifdef LOCAL
    cout << endl;
    }
    #endif // LOCAL
    return 0;
}

  Thank you!

                                                                                                                                             ------from ProLights