给定一个矩阵 A, 返回 A 的转置矩阵。

矩阵的转置是指将矩阵的主对角线翻转,交换矩阵的行索引与列索引。

示例 1:

输入:[[1,2,3],[4,5,6],[7,8,9]]
输出:[[1,4,7],[2,5,8],[3,6,9]]

示例 2:

输入:[[1,2,3],[4,5,6]]
输出:[[1,4],[2,5],[3,6]]

提示:
  1. 1 <= A.length <= 1000
  2. 1 <= A[0].length <= 1000

解析:

    这是一个非常简单的题目,但是我还是做了好几遍才做对,实在是惭愧,我一开始只考虑的方阵,所以就在原矩阵的基础上转置,后来才发现了问题。这个题目原矩阵是row行col列,则需要构建一个col行row列的矩阵作为转置后的矩阵,然后依次赋值,就可以得到结果。

代码:

vector<vector<int>> transpose(vector<vector<int>>& A) 
{
	int row = A.size();
	int col = 0;
	//按照原题,这里可以不用判断
	if (row > 0)
		col = A[0].size();
	else
		return A;
	//构建一个新矩阵
	vector<vector<int>>result = vector<vector<int>>(col, vector<int>(row));
	for (int i = 0; i < row; i++)
	{
		for (int j = 0; j < col; j++)
		{
			result[j][i] = A[i][j];
		}
	}
	return result;
}

 

收藏 打印