sort函數

ACM題的時候,排序是一種經常要用到的操作。如果每次都自己寫個冒泡之類的O(n^2)排序,不但程序容易超時,而且浪費寶貴的比賽時間,還很有可能寫錯。STL裏面有個sort函數,可以直接對數組排序,複雜度爲n*log2(n)。使用這個函數,需要包含頭文件。
 這個函數可以傳兩個參數或三個參數。第一個參數是要排序的區間首地址,第二個參數是區間尾地址的下一地址。也就是說,排序的區間是[a,b)。簡單來說,有一個數組int a[100],要對從a[0]a[99]的元素進行排序,只要寫sort(a,a+100)就行了,默認的排序方式是升序。
 拿我出的AC的策略這題來說,需要對數組t的第0len-1的元素排序,就寫sort(t,t+len);
 對向量v排序也差不多,sort(v.begin(),v.end());
 排序的數據類型不侷限於整數,只要是定義了小於運算的類型都可以,比如字符串類string
 如果是沒有定義小於運算的數據類型,或者想改變排序的順序,就要用到第三參數——比較函數。比較函數是一個自己定義的函數,返回值是bool型,它規定了什麼樣的關係纔是小於。想把剛纔的整數數組按降序排列,可以先定義一個比較函數cmp
bool cmp(int a,int b)
{
 return a>b;
}
 排序的時候就寫sort(a,a+100,cmp);

假設自己定義了一個結構體node
struct node{
 int a;
 int b;
 double c;
}
 有一個node類型的數組node arr[100],想對它進行排序:先按a值升序排列,如果a值相同,再按b值降序排列,如果b還相同,就按c降序排列。就可以寫這樣一個比較函數:
以下是代碼片段:
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;
排序時寫sort(arr,a+100,cmp);

最後看一個完整的實例,初賽時的一道題目文件名排序 
以下是代碼片段:
#include<iostream>
#include<algorithm>
#include<string>
using namespace std;
//定義一個結構體來表示文件,a代表文件名,b代表文件類型(要麼"File"要麼"Dir"
struct node{
 string a,b;
};
//ASCII碼中,所有大寫字母排在所有小寫字母前面,'A'<'Z'<'a'<'z'
//而這題要求忽略大小寫,所以不能直接用字符串的比較。自定義了一個lt函數,就是less than的意思
//先把兩個字符串全部轉化爲小寫,再比較大小(字典序)
bool lt(string x,string y)
{
 int i;
 for(i=0;i<x.length();i++)
 if(x[i]>='A'&&x[i]<='Z')
 x[i]='a'+(x[i]-'A');
 for(i=0;i<y.length();i++)
 if(y[i]>='A'&&y[i]<='Z')
 y[i]='a'+(y[i]-'A');
 return x<y;
}
//自定義的比較函數,先按b值升序排列(也就是"Dir"排在"File"前面)
//如果b值相同,再按a升序排列,用的是剛纔定義的lt函數
bool comp(node x,node y)
{
 if(x.b!=y.b)return x.b<y.b;
 return lt(x.a,y.a);
}
int main()
{
 node arr[10001];
 int size=0;
 while(cin>>arr[size].a>>arr[size].b)
 size++;
 sort(arr,arr+size,comp);
 for(int i=0;i<size;i++)
 cout<<arr[i].a<<" "<<arr[i].b<<endl;
 return 0;
}

發佈了14 篇原創文章 · 獲贊 2 · 訪問量 2萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章