C++,關於二維Vector陣列,以任意列的資料進行排序。
阿新 • • 發佈:2019-01-26
問題描述:存在一個二維vector陣列,例如vector<vector<int>> VIA。二維陣列VIA,VIA.size() = 10,VIA[0].size() = 3;用矩陣表述就是一個十行三列的矩陣。假如陣列想用第一列的資料大小進行排序,用C++函式sort()只需要採用如下程式碼:
sort(viA.begin(), viA.end());//預設為從小到大排序
sort()函式,預設的是對二維陣列按照第一列進行排序。VIA[i][0],這一列。
如果想用第二列或者第三列就必須通過重寫第三引數的輸入規則。這裡有兩種改寫方式:
sort(viA.begin(), viA.end(), [](const vector<int> &a, const vector<int> &b) {return a[0] < b[0]; });
或者:
bool cmp1(const vector<int> &a, const vector<int> &b)
{
return a[0] > b[0];
}
sort(viA.begin(), viA.end(), cmp1);
假如在用第二種方法時出現“本地函式定義是非法的錯誤”,把上面的函式移到mian()函式外面就能解決。
完整的程式碼如下:
#include<stdio.h> #include<algorithm> #include<vector> #include<stdlib.h> #include<iostream> using namespace std; bool cmp1(const vector<int> &a, const vector<int> &b) { return a[0] > b[0]; } int main() { vector<vector<int>> viA(10); for (int i = 0; i < 10; i++) { for (int j = 0; j < 3; j++) { viA[i].push_back(rand() % 100); } } for (int i = 0; i < 10; i++) { for (int j = 0; j < 3; j++) { cout << viA[i][j] << "\t"; } cout << endl; } cout << "按行排序後的輸出" << endl; //sort(viA.begin(), viA.end(), [](const vector<int> &a, const vector<int> &b) {return a[0] < b[0]; }); sort(viA.begin(), viA.end(), cmp1); //for (int i = 0; i < 10; i++) //{ //sort(viA.begin(), viA.end());//預設為從小到大排序 //} for (int i = 0; i < 10; i++) { for (int j = 0; j < 3; j++) { cout << viA[i][j] << "\t"; } cout << endl; } while (1); return 0; }