题目链接:UESTC 1599 wtmsb
题意:给你一组数,每一次取出两个最小的数,将这两个数的和放入这组数中,直到这组数只剩下一个,求最后剩下那个数的大小!
分析:比赛的时候首先我就看到这道题数据是200000,跑时100ms,我把思路捋了一遍,然后讲给旁边人听,一眼看过去,lfh说用哈夫曼树吧,然后找了个板子直接扔上去了,
结果显示出这样的操作:Restricted Function on test 1,从来没见过这个错误啊,百度找了一下,发现这个错误是它oj本身不支持某些函数,比如qsort这种排序操作,wc。。。有毒吧!然后气炸的我扔了一个sort排序,这回总应该过了吧,Time Limit Exceeded on test 1,好的吧,这个怕是有毒哦,后来稳了下yzx,他说优先队列+排序可过,我问:时间复杂度呢!然后,,,,他说不知道,随便搞搞就过了!这TM也行?看来要好好学数据结构了!
优先队列+排序的做法就是取出队头的两个数,然后把它们的和推入到这个队列中,然后优先队列会自动进行排序操作,所以你就能每一次取出两个最小的数,然后推入这两个数的和到队列中去,时间复杂度估计不会超过O(n)(毕竟只有100ms啊,大概也就只能跑1e7的数据)
下面给出AC代码:
1 #include <bits/stdc++.h> 2 using namespace std; 3 const int N=200020; 4 int main() 5 { 6 int n; 7 cin>>n; 8 priority_queue<int, vector<int>, greater<int> >q; 9 for(int i=0;i<n;i++) 10 { 11 int x; 12 cin>>x; 13 q.push(x); 14 } 15 int sum,ans=0; 16 while(1) 17 { 18 int head1=q.top(); 19 q.pop(); 20 int head2=q.top(); 21 q.pop(); 22 sum=head1+head2; 23 ans=ans+sum; 24 if(q.empty()) 25 break; 26 q.push(sum); 27 } 28 cout<<ans<<endl; 29 return 0; 30 }