867. Transpose Matrix

Input: matrix = [[1,2,3],[4,5,6],[7,8,9]]
Output: [[1,4,7],[2,5,8],[3,6,9]]
Input: matrix = [[1,2,3],[4,5,6]]
Output: [[1,4],[2,5],[3,6]]
class Solution {
public:
vector<vector<int>> transpose(vector<vector<int>>& matrix) {

int row = matrix.size();
int column = matrix[0].size();

vector<vector<int>> ans(column, vector<int>(row, 0));

for(int i=0; i<row; i++){

for(int j=0; j<column; j++){

ans[j][i]=matrix[i][j];
}
}
return ans;
}
};

--

--

Get the Medium app

A button that says 'Download on the App Store', and if clicked it will lead you to the iOS App store
A button that says 'Get it on, Google Play', and if clicked it will lead you to the Google Play store