题目
给你一个二维整数数组 matrix, 返回 matrix 的 转置矩阵 。
矩阵的 转置 是指将矩阵的主对角线翻转,交换矩阵的行索引与列索引。
示例1:
输入:matrix = [[1,2,3],[4,5,6],[7,8,9]]
输出:[[1,4,7],[2,5,8],[3,6,9]]
示例2:
输入:matrix = [[1,2,3],[4,5,6]]
输出:[[1,4],[2,5],[3,6]]
提示:
解答方案
因为矩阵的行列数不一定相同,所以不能在原地进行矩阵的转置,必须创建新的矩阵。
class Solution {
public:
vector<vector<int>> transpose(vector<vector<int>>& matrix) {
int m = matrix.size();
int n = matrix[0].size();
vector<vector<int>> res(n, vector<int>(m, 0));//行列数互换,定义一个新的二维数组
//将矩阵数据换至新矩阵的对应位置,即矩阵转置
for (int i = 0; i < m; ++i) {
for (int j = 0; j < n; ++j) {
res[j][i] = matrix[i][j];
}
}
return res;
}
};
总结
vector容器 二维数组两种定义方法
//方法一
#include <string.h>
#include <vector>
#include <iostream>
#include <algorithm>
using namespace std;
int main()
{
int N=5, M=6;
vector<vector<int> > res(N); //定义二维动态数组大小5行
for(int i =0; i< res.size(); i++)//动态二维数组为5行6列,值全为0 默认值为0
{
res[i].resize(M);
}
for(int i=0; i< res.size(); i++)//输出二维动态数组
{
for(int j=0; j<res[i].size(); j++)
{
cout<<res[i][j]<<" ";
}
cout<<"\n";
}
return 0;
}
//方法二
#include <string.h>
#include <vector>
#include <iostream>
#include <algorithm>
using namespace std;
int main()
{
int N=5, M=6;
vector<vector<int> > res(N, vector<int>(M)); //定义二维动态数组5行6列
for(int i=0; i< res.size(); i++)//输出二维动态数组
{
for(int j=0;j<res[i].size();j++)
{
cout<<res[i][j]<<" ";
}
cout<<"\n";
}
return 0;
}
resize()函数
vector.resize(int n,element)表示调整容器vector的大小为n,扩容后的每个元素的值为element,默认为0
resize()会改变容器的容量和当前元素个数
resize()并不会对原vector已经存在的元素进行重新初始化!!!