合併果子(貪心+優先佇列)
阿新 • • 發佈:2019-02-12
現在有n堆果子,第i堆有ai個果子。現在要把這些果子合併成一堆,每次合併的代價是兩堆果子的總果子數。求合併所有果子的最小代價。
Input
第一行包含一個整數T(T<=50),表示資料組數。
每組資料第一行包含一個整數n(2<=n<=1000),表示果子的堆數。
第二行包含n個正整數ai(ai<=100),表示每堆果子的果子數。
Output
每組資料僅一行,表示最小合併代價。
Sample Input
2 4 1 2 3 4 5 3 5 2 1 4
Sample Output
19 33
分析:
貪心的思想
每次選擇價值最小和次小的那兩個進行合併,合併成新的果子放進果堆中,然後又在果堆中選擇最小的和次小的進行合併
這樣下來,合併所有果子所花費的代價肯定是最小的
所以我們可以用優先佇列來實現這個過程
代價最小的果子優先
code:
#include<stdio.h> #include <iostream> #include <math.h> #include <queue> using namespace std; int main() { int t,n,a; scanf("%d",&t); while(t--) { priority_queue<int,vector<int>,greater<int> >q;//優先佇列,數值小的優先 scanf("%d",&n); for(int i=0;i<n;i++) { scanf("%d",&a); q.push(a); } int ans=0,temp; while(q.size()>=2)//果堆中只有一個果子的時候合併就完成了 { temp=0; temp+=q.top(),q.pop(); temp+=q.top(),q.pop();//每次去優先佇列隊首的兩個,因為他們是代價最小和次小的 ans+=temp; q.push(temp);//將合併後的果子再次放入堆中 } printf("%d\n",ans); } return 0; }