關於C++中vector和set使用sort方法進行排序

C++中vector和set都是非常方便的容器，

sort方法是algorithm頭文件里的一個標准函數，能進行高效的排序，默認是按元素從小到大排序

將sort方法用到vector和set中能實現多種符合自己需求的排序

首先sort方法可以對靜態的數組進行排序

#include<iostream>
using namespace std;
int main(){
    int a[10] = { 9, 0, 1, 2, 3, 7, 4, 5, 100, 10 };
    sort(a, a +10);
    for (int i = 0; i < 10; i++)
        cout << a[i] << endl;
    return 0;
}

運行結果：

 

這里可以看到是sort(a,a+10)，但是數組a一共只有9個元素，為什么是a+10而不是a+9呢？

因為sort方法實際上最后一位地址對應的數是不取的，

而且vector，set，map這些容器的end()取出來的值實際上並不是最后一個值，而end的前一個才是最后一個值！

需要用prev(xxx.end())，才能取出容器中最后一個元素。

 

對vector使用sort函數：

第一種情形：基本類型,如vector<int>,vector<double>,vector<string>也是可以的

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
int main(){
    vector<int> a;
    int n = 5;
    while (n--){
        int score;
        cin >> score;
        a.push_back(score);
    }
    //cout <<" a.end()"<< *a.end() << endl;       執行這句話會報錯！
    cout << " prev(a.end)" << *prev(a.end()) << endl;
    sort(a.begin(), a.end());
    for (vector<int>::iterator it = a.begin(); it != a.end(); it++){
        cout << *it << endl;
    }
    return 0;
}

執行結果：

看到了嗎，實際上end的前一個指針指向的元素才是插入時的最后一個值！

排序后從小大大。

第二種情形：用自定義的結構體進行sort算法，

這時候需要自己定義個比較函數，因為sort算法是基於容器中的元素是可以兩兩比較的，然后從小到大排序，所以要自定義怎么樣才是小於（'<'）

#include<iostream>
#include<vector>
#include<set>
#include<string>
#include<algorithm>
using namespace std;
struct student{
    char name[10];
    int score;
};
//自定義“小於”
bool comp(const student &a, const student &b){
    return a.score < b.score;
}
int main(){
    vector<student> vectorStudents;
    int n = 5;
    while (n--){
        student oneStudent;
        string name;
        int score;
        cin >> name >> score;
        strcpy(oneStudent.name, name.c_str());
        oneStudent.score = score;
        vectorStudents.push_back(oneStudent);
    }
    cout << "===========排序前================" << endl;
    for (vector<student>::iterator it = vectorStudents.begin(); it != vectorStudents.end(); it++){
        cout << "name: " << it->name << " score: " << it->score << endl;
    }
    sort(vectorStudents.begin(),vectorStudents.end(),comp);
    cout << "===========排序后================" << endl;
    for (vector<student>::iterator it = vectorStudents.begin(); it != vectorStudents.end(); it++){
        cout << "name: " << it->name << " score: " << it->score << endl;
    }
    return 0;
}

運行結果：

 

不過有時候一個排序條件不夠，比如要求學生按分數從高到低排序，如果分數相同，則按照年齡從大到小排序

就需要在comp自定義函數里面修改一下判斷了，原來是直接return a.score < b.score

現在就需要判斷

if (a.score > b.score)
   return true; else if (a.score == b.score && a.age > b.age)
 return true; else return false;

這里一定要記得else return false！！！有一次比賽的時候寫到這個函數，有三個判斷條件，結果忘了這茬，總是報錯，
到后來有點着急了就自己手動實現了一下寫了三個比較函數，調用了三次sort函數！！！！！

#include<iostream>
#include<vector>
#include<set>
#include<string>
#include<algorithm>
using namespace std;
struct student{
    char name[10];
    int score;
    int age;
};
//自定義“小於”
bool comp(const student &a, const student &b){
    if (a.score > b.score)
        return true;
    else if (a.score == b.score  && a.age > b.age)
        return true;
    else                ///這里的else return false非常重要！！！！！
        return false;
}
int main(){
    vector<student> vectorStudents;
    /*set<student> setStudents;*/
    //int n = 5;
    int n = 6;
    while (n--){
        student oneStudent;
        string name;
        int score;
        int age;
        cin >> name >> score>>age;
        strcpy(oneStudent.name, name.c_str());
        oneStudent.score = score;
        oneStudent.age = age;
        vectorStudents.push_back(oneStudent);
    }
    cout << "===========排序前================" << endl;
    for (vector<student>::iterator it = vectorStudents.begin(); it != vectorStudents.end(); it++){
        cout << "name: " << it->name << " score: " << it->score << " age: "<<it->age<<endl;
    }
    sort(vectorStudents.begin(), vectorStudents.end(), comp);
    //sort(setStudents.begin(), setStudents.end());
    cout << "===========排序后================" << endl;
    for (vector<student>::iterator it = vectorStudents.begin(); it != vectorStudents.end(); it++){
        cout << "name: " << it->name << " score: " << it->score << " age: " << it->age << endl;
    }
    return 0;
}

運行結果如下：

接下來，對於set做類似的操作。

set是一個集合，內部的元素不會重復，同時它會自動進行排序，也是從小到大

而且set的insert方法沒有insert(a,cmp)這種重載，所以如果要把結構體插入set中，我們就要重載'<'運算符。

set方法在插入的時候也是從小到大的，那么我們重載一下<運算符讓它從大到小排序

#include<iostream>
#include<vector>
#include<set>
#include<string>
#include<algorithm>
using namespace std;
struct student{
    char name[10];
    int score;
};
//自定義“小於”
bool comp(const student &a, const student &b){
    return a.score < b.score;
}
bool operator < (const student & stu1,const student &stu2){
    return stu1.score > stu2.score;
}
int main(){
    //vector<student> vectorStudents;
    set<student> setStudents;
    //int n = 5;
    int n = 6;
    while (n--){
        student oneStudent;
        string name;
        int score;
        cin >> name >> score;
        strcpy(oneStudent.name, name.c_str());
        oneStudent.score = score;
        setStudents.insert(oneStudent);
    }
    cout << "===========排序前================" << endl;
    for (set<student>::iterator it = setStudents.begin(); it != setStudents.end(); it++){
        cout << "name: " << it->name << " score: " << it->score << endl;
    }
    //sort(setStudents.begin(), setStudents.end(), comp);
    //cout << "===========排序后================" << endl;
    //for (set<student>::iterator it = setStudents.begin(); it != setStudents.end(); it++){
    //    cout << "name: " << it->name << " score: " << it->score << endl;
    //}
    return 0;
}

運行結果：

我們可以看到，set內元素不會重復，而且它按照它所認為的“從小到大”進行了排序