欢迎访问==> 【考研OR保研】机试题
矩阵对角线 是一条从矩阵最上面行或者最左侧列中的某个元素开始的对角线,沿右下方向一直到矩阵末尾的元素。例如,矩阵 mat
有 6
行 3
列,从 mat[2][0]
开始的 矩阵对角线 将会经过 mat[2][0]
、mat[3][1]
和 mat[4][2]
。
给你一个 m * n
的整数矩阵 mat
,请你将同一条 矩阵对角线 上的元素按升序排序后,返回排好序的矩阵。
示例 1:
输入:mat = [[3,3,1,1],[2,2,1,2],[1,1,1,2]]
输出:[[1,1,1,1],[1,2,2,2],[1,2,3,3]]
示例 2:
输入:mat = [[11,25,66,1,69,7],[23,55,17,45,15,52],[75,31,36,44,58,8],[22,27,33,25,68,4],[84,28,14,11,5,50]]
输出:[[5,17,4,1,52,7],[11,11,25,45,8,69],[14,23,25,44,58,15],[22,27,31,36,50,66],[84,28,75,33,55,68]]
提示:
m == mat.length
n == mat[i].length
1 <= m, n <= 100
1 <= mat[i][j] <= 100
C++代码
class Solution {
public:
vector<vector<int>> diagonalSort(vector<vector<int>>& mat) {
int n = mat.size(), m = mat[0].size();
vector<vector<int>> res(n, vector<int>(m));
for(int i = 0; i < n; i ++)
{
for(int j = 0; j < m; j ++)
{
if(i == 0 || j == 0) //对角线的起始
{
vector<int> t;
for(int k = 0; k + i < n && k + j < m; k ++) //对角线
{
t.push_back(mat[k + i][k + j]); //取出该对角线内的元素
}
sort(t.begin(), t.end()); //排序
for(int k = 0; k + i < n && k + j < m; k ++) //重新填写该对角线
{
res[i + k][j + k] = t[k];
}
}
}
}
return res;
}
};
击败了98%用户 写的思路太好了 NB
orz