1. 程式人生 > >C++,關於二維Vector陣列,以任意列的資料進行排序。

C++,關於二維Vector陣列,以任意列的資料進行排序。

問題描述:存在一個二維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;
}