簡單題 這次用c++ stl 的模板做的 但是效率很低啊 600Ms 過的
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn = 3000+10;
int data[maxn];
int re[maxn*maxn/2];
bool cmp(int a,int b)
{
return a>b;
}
int main()
{
int N,M,ans;
while(scanf("%d%d",&N,&M)!=EOF)
{
for(int i = 0; i < N; i ++)
scanf("%d",&data[i]);
//sort(data,data+N,cmp);
ans = 0;
for( i = 0;i < N;i++)
for(int j = i+1;j<N;j++)
{
re[ans++] = data[i]+data[j];
}
sort(re,re+N*(N-1)/2,cmp);
for( i = 0; i < M; i ++)
printf(i ==0 ?"%d":" %d",re[i]);
printf("\n");
}
return 0;
}
摘自
hi.baidu.com/topman3758/blog/item/859b18da6222623933fa1cd8.html
做ACM題的時候,排序是一種經(jīng)常要用到的操作。如果每次都自己寫個冒泡之類的O(n^2)排序,不但程序容易超時,而且浪費(fèi)寶貴的比賽時間,還很有可能寫錯。STL里面有個sort函數(shù),可以直接對數(shù)組排序,復(fù)雜度為n*log2(n)。使用這個函數(shù),需要包含頭文件。
這個函數(shù)可以傳兩個參數(shù)或三個參數(shù)。第一個參數(shù)是要排序的區(qū)間首地址,第二個參數(shù)是區(qū)間尾地址的下一地址。也就是說,排序的區(qū)間是[a,b)。簡單來說,有一個數(shù)組int a[100],要對從a[0]到a[99]的元素進(jìn)行排序,只要寫sort(a,a+100)就行了,默認(rèn)的排序方式是升序。
拿我出的“AC的策略”這題來說,需要對數(shù)組t的第0到len-1的元素排序,就寫sort(t,t+len);
對向量v排序也差不多,sort(v.begin(),v.end());
排序的數(shù)據(jù)類型不局限于整數(shù),只要是定義了小于運(yùn)算的類型都可以,比如字符串類string。
如果是沒有定義小于運(yùn)算的數(shù)據(jù)類型,或者想改變排序的順序,就要用到第三參數(shù)——比較函數(shù)。比較函數(shù)是一個自己定義的函數(shù),返回值是bool型,它規(guī)定了什么樣的關(guān)系才是“小于”。想把剛才的整數(shù)數(shù)組按降序排列,可以先定義一個比較函數(shù)cmp
bool cmp(int a,int b)
{
return a>b;
}
排序的時候就寫sort(a,a+100,cmp);
假設(shè)自己定義了一個結(jié)構(gòu)體node
struct node{
int a;
int b;
double c;
}
有一個node類型的數(shù)組node arr[100],想對它進(jìn)行排序:先按a值升序排列,如果a值相同,再按b值降序排列,如果b還相同,就按c降序排列。就可以寫這樣一個比較函數(shù):
以下是代碼片段:
bool cmp(node x,node y)
{
if(x.a!=y.a) return x.a
if(x.b!=y.b) return x.b>y.b;
return return x.c>y.c;
}
posted on 2010-05-24 23:52
付翔 閱讀(364)
評論(0) 編輯 收藏 引用 所屬分類:
ACM 數(shù)據(jù)結(jié)構(gòu) 、
c++