題目的算法就是每次選擇水果排序後,每次選擇重量最小的兩堆合併
#include<iostream>
#include<vector>
#include<iterator>
#include<set>
using namespace std;
int main()
{
int N;
cin>>N;
while(N--)
{
int num;
cin>>num;
multiset<long long>Apple;
while(num--)
{ long long weight;
cin>>weight;
Apple.insert(weight);
}
multiset<int>::iterator it;
long long temp1,temp2,result=0;
while(Apple.size()!=1)
{
temp1=*Apple.begin(); //set 是自動由小到大排序所以只要取出第一第二個即可
Apple.erase(Apple.begin());
temp2=*Apple.begin();
Apple.erase(Apple.begin());
result+=temp1+temp2;
Apple.insert(temp1+temp2);
}
cout<<result<<endl;
}
}
注:不要使用algorithm頭文件中的sort排序,因爲這個在ACM平臺上回超時,同時最好也別使用multiset.erase(first, last )貌似這個也會超時。。。很蛋疼。
最好請大神幫我分析一下set得數據結構,以及自帶排序得實現和時間複雜度,因爲sort是nlogn