19.顺时针打印矩阵
2019-08-12 本文已影响0人
iwtbam
题目描述
- 输入一个矩阵,按照从外向里以顺时针的顺序依次打印出每一个数字,例如,如果输入如下4 X 4矩阵: 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 则依次打印出数字1,2,3,4,8,12,16,15,14,13,9,5,6,7,11,10.
解题思路
- 定义好打印的方向及先右,再下,然后左,最后上的顺序, 碰到边界,或者已访问的元素即切换按顺序切换方向。
AC代码
class Solution {
public:
vector<int> printMatrix(vector<vector<int> > matrix) {
int rows = matrix.size();
int cols = matrix[0].size();
vector<vector<int>> label(rows, vector<int>(cols, 0));
//vector<vector<int>> label = {rows, vector<int>(cols, 0)}; 其实我更喜欢这样写可惜牛客网的编译器不支持C++17
int dx[] = { 1, 0, -1, 0 };
int dy[] = { 0, 1, 0 ,-1 };
int dir = 0;
int curx = 0;
int cury = 0;
vector<int> iv;
int total = rows * cols;
iv.push_back(matrix[0][0]);
label[0][0] = 1;
while (iv.size() < total) {
int nextx = curx + dx[dir];
int nexty = cury + dy[dir];
if (nextx >= 0 && nextx < cols && nexty >= 0 && nexty < rows&& !label[nexty][nextx]) {
curx = nextx;
cury = nexty;
label[nexty][nextx] = 1;
iv.push_back(matrix[cury][curx]);
}
else {
dir = (dir + 1) % 4;
}
}
return iv;
}
};