C++,关于二维Vector数组,以任意列的数据进行排序。
程序员文章站
2022-03-22 13:33:27
...
问题描述:存在一个二维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;
}
如果还有疑惑请参考这个人的博客,https://www.cnblogs.com/grandyang/p/4843528.html
上一篇: HDU1027
下一篇: typedef 以及vector的应用