还记得Gardon给小希布置的那个作业么?(上次比赛的1005)其实小希已经找回了原来的那张数表,现在她想确认一下她的答案是否正确,但是整个的答案是很庞大的表,小希只想让你把答案中最大的M个数告诉她就可以了。
给定一个包含N(N<=3000)个正整数的序列,每个数不超过5000,对它们两两相加得到的N*(N-1)/2个和,求出其中前M大的数(M<=1000)并按从大到小的顺序排列。
Input输入可能包含多组数据,其中每组数据包括两行:
第一行两个数N和M,
第二行N个数,表示该序列。
Output对于输入的每组数据,输出M个数,表示结果。输出应当按照从大到小的顺序排列。Sample Input
4 4
1 2 3 4
4 5
5 3 6 4
Sample Output
7 6 5 5
11 10 9 9 8
第一次给别人用vector讲这道题的时候(顺便想宣传下vector的好处),没想到“Memory Limit Exceeded”(内存超限);
好尴尬,这是暴力代码:
#include<iostream> //记住,选C++编译器,而不是GCC/G++
#include<vector>
#include<algorithm>
using namespace std;
int a[3010];
bool cmp(int a,int b){
return a>b;
}
vector<int> v;
int main(){
int m,n,t,i,j;
while(cin>>m>>n&&m&&n){
for(i=0;i<m;i++){
cin>>a[i];
}
for(i=0;i<m-1;i++){
for(j=i+1;j<m;j++){
t=a[i]+a[j];
v.push_back(t);
}
}
sort(v.begin(),v.end(),cmp);
for(i=0;i<n-1;i++){
cout<<v[i]<<" ";
}
cout<<v[n-1]<<endl;
v.clear();
v.shrink_to_fit();
}
return 0;
}
问题就出在vector内存的释放上面;ps:(上面的v.clear();v.shrink_to_fit()可以换成vector<int>().swap(v); );
一开始选的G++编译器;查资料发现,这两种都是C++11标准支持的,而有些G++编译器不支持C++11标准(有些GCC编译器甚至编译都通不过,如DEV),所以说,这两种用G++提交不会报错,但实际并不会执行,内存释放不掉,造成超时。
vector<int> ().swap(v);尽管在DEV编译能通过,但某oj的GCC编译器似乎也不和DEV一样,
最后还是选择C++编译器AC了,哎,松一口气~~
再加一种,哈希:
#include<iostream>
#include<cstring>
using namespace std;
int a[3500];
int cnt[100001];
int main(){
int n,m;
while(cin>>n>>m){
memset(cnt,0,sizeof(cnt));
for(int i=0;i<n;i++){
cin>>a[i];
}
for(int i=0;i<n-1;i++){
for(int j=i+1;j<n;j++){
cnt[a[i]+a[j]]++;
}
}
for(int i=10000;m>0;){
if(cnt[i]&&m>1){
cout<<i<<" ";
m--,cnt[i]--;
}
else if(cnt[i]&&m==1){
cout<<i<<endl;
cnt[i]--,m--;
}
else i--;
}
}
return 0;
}