C++学习笔记 —— STL之二维数组

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
// g++ -std=c++11 2Dvector.cc -o test && test.exe
//遍历数组一： 直接访问
void traverseVecvec1(vector<vector<int>> vecvec)
{
    
    
    if (vecvec.empty()) {
    
    
        cout << "The vector is empty!" << endl;
        return;
    }
    for (int i = 0; i < vecvec.size(); i++) {
    
    
        for (int j = 0; j < vecvec[i].size(); j++) {
    
    
            cout << vecvec[i][j] << " ";
        }
        cout << endl;
    }
    cout << endl;
}
// 遍历数组二： 使用迭代器
void traverseVecvec2(vector<vector<int>> vecvec)
{
    
    
    if (vecvec.empty()) {
    
    
        cout << "The vector is empty!" << endl;
        return;
    }
    // vector<vector<int>>::iterator iter;
    for (auto iter = vecvec.begin(); iter != vecvec.end(); iter++) {
    
    
        //(*iter)表示每一行的vector
        for (int i = 0; i < (*iter).size(); ++i) {
    
    
            cout << (*iter)[i] << " ";
        }
        cout << endl;
    }
    cout << endl;
}

//比较函数，自定义比较函数，比较的是两个一维数组。
static bool cmp(const vector<int>& a, const vector<int>& b)
{
    
    
    if (a[0] > b[0])  //按照第一个数组第一个元素比第二个数组第一个元素大进行排列
        return true;
    if (a[0] == b[0] && a[1] < b[1])  //如果第一个数组一元素与第二个数组一元素相等，则按照第二个元素小进行排列
        return true;

    return false;
}
int main()
{
    
    
    //初始化一
    vector<vector<int>> vecvec1;  //定义一个二维数组为空
    // 赋值
    //二维数组行赋值
    vector<int> vec1;  //第一行
    vec1.push_back(1);
    vec1.push_back(2);
    vec1.push_back(3);
    vector<int> vec2;  //第二行
    vec2.push_back(4);
    vec2.push_back(5);
    vec2.push_back(6);
    vec2.push_back(7);
    //把每行放入到二维数组中
    vecvec1.push_back(vec1);
    vecvec1.push_back(vec2);
    traverseVecvec1(vecvec1);
    //初始化二：（初始化5行3列）矩阵式
    int row = 5;
    int col = 3;
    vector<vector<int>> vecvec2(row, vector<int>(col));  //定义一个二维数组有row行，每行有col个元素，初始化为0
    // 可以每个元素赋值
    for (int i = 0; i < row; i++) {
    
    
        for (int j = 0; j < col; j++) {
    
    
            vecvec2[i][j] = (i + 1) * (j + 1);
        }
    }
    traverseVecvec1(vecvec2);

    //初始化方式三：
    //输入每行要传入的元素
    vector<vector<int>> vecvec3;
    cout << "please input row number: " << endl;
    int row2 = 0;
    cin >> row2;
    for (int i = 0; i < row2; i++) {
    
    
        cout << "please input numbers in "<< i << " line."<< endl;
        vector<int> r;  //定义临时数组存储每一行
        int n = 0;      //临时变量作为每行有几个数
        cin >> n;
        for (int j = 0; j < n; j++) {
    
    
            r.push_back(9);
        }
        vecvec3.push_back(r);
    }
    traverseVecvec2(vecvec3);

    //排序
    sort(vecvec3.begin(), vecvec3.end(), cmp);
}