1. 程式人生 > >C++ sort vector > or vector 容器的排序

C++ sort vector > or vector 容器的排序

C++的STL中提供了很強大的排序函式sort,可以對任意陣列,結構體及類進行排序,下面我們先來看最簡單的陣列排序。預設的升序排列,我們也可以在後面加上less或greater來告訴編譯器我們想要的排序順序。

vector<int> v = {2, 0, 1, 5, 9, 2, 7};

// Ascending order
sort(v.begin(), v.end());
sort(v.begin(), v.end(), less<int>());

// Descending order
sort(v.rbegin(), v.rend());
sort(v.begin(), v.end(), greater
<int>());

如果是一個二維陣列,也可以是用sort,我們可以選擇根據某一列來進行排序,如果我們不重寫cmp函式,那麼預設的是根據第一列來排序,當然我們可以通過重寫來根據其他列來排序:

/* Input matrix
m = [
        1 4 2 
        0 8 3
        3 5 1
    ]
*/

// Ascending order by first column
sort(m.begin(), m.end());
/*
m = [
        0 8 3
        1 4 2 
        3 5 1
    ]
*/ // Descending order by first column sort(m.rbegin(), m.rend()); /* m = [ 3 5 1 1 4 2 0 8 3 ] */ // Ascending order by second column sort(m.begin(), m.end(), [](const vector<int> &a, const vector<int> &b) { return a[1] < b[1]; } ); bool
cmp(const vector<int> &a, const vector<int> &b) { return a[0] > b[0]; } sort(m.begin(), m.end(), cmp); /* m = [ 1 4 2 3 5 1 0 8 3 ] */ // Descending order by second column sort(m.begin(), m.end(), [](const vector<int> &a, const vector<int> &b) { return a[1] > b[1]; } ); bool cmp(const vector<int> &a, const vector<int> &b) { return a[0] < b[0]; } sort(m.begin(), m.end(), cmp); /* m = [ 0 8 3 3 5 1 1 4 2 ] */

下面來看如何給class排序,我們將自定義的類class裝入到容器vector中,然後根據class中的一個引數為依據進行排序,這裡分兩種情況,一個是vector中裝指標的class,一種的裝正常的class,它們再重寫cmp比較函式上有些區別的,參見程式碼如下:

class A {
public:
    int a1, a2;
    A(int m, int n): a1(m), a2(n) {}
};

class B {
public:
    int b1, b2;
    B(int m, int n): b1(m), b2(n) {}
};

bool cmp1(A const *a, A const *b) {
    return a->a1 < b->a1;
}

bool cmp2(B const &a, B const &b) {
    return a.b1 < b.b1;
}

void printArray(vector<A*> array) {
    for (int i = 0; i < array.size(); ++i) {
        cout << array[i]->a1 << " " << array[i]->a2 << endl;
    }
    cout << endl;
}

void printArray2(vector<B> array) {
    for (int i = 0; i < array.size(); ++i) {
        cout << array[i].b1 << " " << array[i].b2 << endl;
    }
    cout << endl;
}

int main() {

    vector<A*> array;
    array.push_back(new A(65, 100));
    array.push_back(new A(70, 150));
    array.push_back(new A(56, 90));
    array.push_back(new A(75, 190));
    array.push_back(new A(60, 95));
    array.push_back(new A(68, 110));
    
    printArray(array);
    sort(array.begin(), array.end(), cmp1);
    printArray(array);
    
    vector<B> array2;
    array2.push_back(B(65, 100));
    array2.push_back(B(70, 150));
    array2.push_back(B(56, 90));
    array2.push_back(B(75, 190));
    array2.push_back(B(60, 95));
    array2.push_back(B(68, 110));
    
    printArray2(array2);
    sort(array2.begin(), array2.end(), cmp2);
    printArray2(array2);
    
    return 0;
}