1. 程式人生 > >NyistOJ 55 懶省事的小明(c++)(優先隊列)(貪心)

NyistOJ 55 懶省事的小明(c++)(優先隊列)(貪心)

pri 限制 esp 等於 題解 測試數據 小明 num 成了

懶省事的小明

時間限制:3000 ms | 內存限制:65535 KB

難度:3

描述

小明很想吃果子,正好果園果子熟了。在果園裏,小明已經將所有的果子打了下來,而且按果子的不同種類分成了不同的堆。小明決定把所有的果子合成一堆。 因為小明比較懶,為了省力氣,小明開始想點子了:

每一次合並,小明可以把兩堆果子合並到一起,消耗的體力等於兩堆果子的重量之和。可以看出,所有的果子經過n-1次合並之後,就只剩下一堆了。小明在合並果子時總共消耗的體力等於每次合並所耗體力之和。

因為還要花大力氣把這些果子搬回家,所以小明在合並果子時要盡可能地節省體力。假定每個果子重量都為1,並且已知果子的種類數和每種果子的數目,你的任務是設計出合並的次序方案,使小明耗費的體力最少,並輸出這個最小的體力耗費值。

例如有3種果子,數目依次為1,2,9。可以先將1、2堆合並,新堆數目為3,耗費體力為3。接著,將新堆與原先的第三堆合並,又得到新的堆,數目為12,耗費體力為12。所以小明總共耗費體力=3+12=15。可以證明15為最小的體力耗費值。

輸入

第一行輸入整數N(0<N<=10)表示測試數據組數。接下來每組測試數據輸入包括兩行,第一行是一個整數n(1<=n<=12000),表示果子的種類數。第二行包含n個整數,用空格分隔,第i個整數ai(1<=ai<=20000)是第i種果子的數目。

輸出

每組測試數據輸出包括一行,這一行只包含一個整數,也就是最小的體力耗費值。

樣例輸入

1
3
1 2 9

樣例輸出

15

題解

貪心+優先隊列。強勢占坑!!國慶節補充…

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <map>
#include <queue>
#define ll long long

using namespace std ;

int main(){
    int t ;
    cin >> t ;
    while ( t -- ){
        int
n ; cin >> n ; ll ans = 0 ; priority_queue<ll , vector<ll> , greater<ll> > p_que ; for ( int i = 0 ; i < n ; i ++ ){ int x ; cin >> x ; p_que.push(x) ; } int num = 0 ; while( p_que.size() != 1 ){ num = p_que.top() ; p_que.pop() ; num += p_que.top() ; p_que.pop() ; ans += num ; p_que.push(num) ; } cout << ans << endl ; } return 0 ; }

NyistOJ 55 懶省事的小明(c++)(優先隊列)(貪心)