题目描述:
对于一个矩阵,请设计一个算法从左上角(mat[0][0])开始,顺时针打印矩阵元素。
给定int矩阵mat,以及它的维数nxm,请返回一个数组,数组中的元素为矩阵元素的顺时针输出。
测试样例:
[[1,2],[3,4]],2,2
返回:[1,2,4,3]
代码:
#include<iostream>
#include<vector>
using namespace std;
class Printer
{
public:
vector<int> clockwisePrint(vector<vector<int> > mat, int n, int m)
{
vector<int>res;
int x1 = 0, y1 = 0;//矩阵左上角坐标
int x2 = n - 1, y2 = m - 1;//右下角坐标
//打印一圈后需缩小矩阵大小
while (x1 <= x2 && y1 <= y2)//循环条件
{
for (int j = y1; j <= y2; ++j)
{
res.push_back(mat[x1][j]);
}
for (int i = x1 + 1; i < x2; ++i)
{
res.push_back(mat[i][y2]);
}
for (int j = y2; x1 < x2 && j >= y1; --j)
//只剩一行的话,第一步就已经将它放进res中了
//所以加条件
{
res.push_back(mat[x2][j]);
}
for (int i = x2 - 1; y1 < y2 && i>x1; --i)
//y1<y2 只有在不是单一竖列的时候才执行
{
res.push_back(mat[i][y1]);
}
x1++; y1++;
x2--; y2--;
}
return res;
}
};
链接:https://www.nowcoder.com/questionTerminal/9b4c81a02cd34f76be2659fa0d54342a
来源:牛客网