C++ 函数返回二维数组和二维vector容器 vector<vector<int>>

C++ 一个很头疼的事情就是只能返回一个值,想要返回多个值只能将这些值存在数组中,返回一个数组或者vector。

1、返回二维数组:

#include 
using namespace std;


int **func(int row,int col)
{	
	int **data = 0;     //初始化一个二维数组
	data = new int*[row];   //建立包含row个指针的指针数组
	for (int i = 0; i < row; i++)
	{
		*(data + i) = new int[col];   //每个一维指针数组的长度都是col
	}

	for (int i = 0; i < row; i++)
		for (int j = 0; j < col; j++)
			data[i][j] = i + j;     //随便定义元素

	return data;
}


int main()
{
	int row = 5;
	int col = 8;
	int **data = func(row, col);
	for (int i = 0; i < row; i++)
		for (int j = 0; j < col; j++)
		{
			if (j < col - 1)
				cout << data[i][j] << "\t";
			else
				cout << data[i][j] << endl;
		}		
	return 0;
}

输出:

0       1       2       3       4       5       6       7
1       2       3       4       5       6       7       8
2       3       4       5       6       7       8       9
3       4       5       6       7       8       9       10
4       5       6       7       8       9       10      11

其实这样用指针并不好,容易出现空指针、越界等问题。

2、返回 vector

vector是一个类模板而非类型,vector、vector等才是类型。

关于二维vector的元素插入方式可参考博客:关于C++中vector的使用

使用vector时一定要注意容量和下标的问题,尤其要用下标赋值时,要先确定vector是否为空以及容量是否足够。最安全的方法是组合使用 v.push_back() 、v.pop_back() 和 v.clear()

code:

#include 
#include 
using namespace std;


vector<vector<int>> func(int row,int col)
{
    vector<vector<int>> data;    //二维vector
    vector<int> data_row(col);   //存放每行数据的一维vector
    for(int i=0;i<row;i++)
    {
        for(int j=0;j<col;j++)
            data_row[j]=i+j;     //使用下标对vector赋值时要注意vector容量是否足够
        data.push_back(data_row);
    }
    return data;
}


int main()
{
    int row = 5;
    int col = 8;
    vector<vector<int>> data = func(row, col);
    for (int i = 0; i < row; i++)
        for (int j = 0; j < col; j++)
        {
            if (j < col - 1)
                cout << data[i][j] << "\t";
            else
                cout << data[i][j] << endl;
        }
    return 0;
}

输出:

0       1       2       3       4       5       6       7
1       2       3       4       5       6       7       8
2       3       4       5       6       7       8       9
3       4       5       6       7       8       9       10
4       5       6       7       8       9       10      11

3、二维数组 vs 二维vector

3.1 二维数组

因为C++规定多维数组初始化时一定要给出列数,行数可以缺省,所以二维数组中每个子数组的元素数量一定相同。如果初始化时数量不相同,则用默认值补全。

例子:

#include 
using namespace std;

int main()
{
    int data[][5] = {{1,2,3},
    				{4,5,6,7}, 
    				{8,9,10,11,12}};
    for(auto &row:data)   //注意这里是引用 &row,因为row是个指针
    {
        for(auto col:row)
            cout<<col<<'\t';
        cout<<endl;
    }
    return 0;
}

输出:

1       2       3       0       0
4       5       6       7       0
8       9       10      11      12

可以发现,每行元素数量必须相同。但在实际应用中,多维数组中的每个子数组元素数量并不一定相同,因此数组就显得不那么人性化了,这种情况就可以考虑使用二维 vector。

上面的例子用了C++11的新语法:范围for循环。要注意使用范围for循环对二维数组进行迭代时,第一层循环的 row 要用引用,因为 row 不是普通的数组元素,而是子数组首元素的指针。

3.2 二维 vector : vector

二维 vector 中,每个子vector 的长度可以不相同。如以下例子:

#include 
#include 
using namespace std;

int main()
{
    vector<vector<int>> data = {{1,2,3},
                                {4,5,6,7},
                                {8,9,10,11,12}};
    for(auto row:data)
    {
        for(auto col:row)
            cout<<col<<'\t';
        cout<<endl;
    }
    cout<<"每个子vector的长度:"<<endl;
    for(auto row:data)
        cout<<row.size()<<'\t';
    return 0;
}

输出:

1       2       3
4       5       6       7
8       9       10      11      12
每个子vector的长度:
3       4       5

你可能感兴趣的:(C++,c++,算法)