JZ29. 顺时针打印矩阵
开辟二维bool数组
开辟的二维数组检查是否已经遍历过
空间复杂度需要O(nm)
代码:
#include <vector>
class Solution {
public:
vector<int> printMatrix(vector<vector<int>> matrix) {
vector<int> res;
if (matrix.empty()) return res;
int n = matrix.size(), m = matrix[0].size();
vector<vector<bool>> st(n, vector<bool>(m, false));
int dx[4] = {-1, 0, 1, 0}, dy[4] = {0, 1, 0, -1};
int x = 0, y = 0, d = 1;
for (int k = 0; k < n*m; k ++ ) {
res.push_back(matrix[x][y]);
st[x][y] = true;
int a = x + dx[d], b = y + dy[d];
if (a < 0 || a >= n || b < 0 || b >= m || st[a][b]) {
d = (d + 1) % 4;
a = x + dx[d], b = y + dy[d];
}
x = a, y = b;
}
return res;
}
};
朴素遍历法
JZ31. 栈的压入、弹出序列
题目描述:
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否可能为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1,2,3,4,5是某栈的压入顺序,序列4,5,3,2,1是该压栈序列对应的一个弹出序列,但4,3,5,1,2就不可能是该压栈序列的弹出序列。
代码:
class Solution {
public:
bool IsPopOrder(vector<int>& pushV, vector<int>& popV) {
// write code here
if (pushV.size() != popV.size()) return false;
stack<int> stk;
int i = 0;
for (auto x : pushV) {
stk.push(x);
while (stk.size() && stk.top() == popV[i]) {
stk.pop();
i ++ ;
}
}
return stk.empty();
}
};
时间复杂度O(n)
JZ32. 从上往下打印二叉树
题目描述:
不分行从上往下打印出二叉树的每个节点,同层节点从左至右打印。例如输入{8,6,10,#,#,2,1},如以下图中的示例二叉树,则依次打印8,6,10,2,1(空节点不打印,跳过),请你将打印的结果存放到一个数组里面,返回。
代码:
class Solution {
public:
vector<int> PrintFromTopToBottom(TreeNode* root) {
vector<int> res;
if (!root) return res;
queue<TreeNode*> q;
q.push(root);
while (q.size()) {
auto t = q.front();
q.pop();
res.push_back(t->val);
if (t->left) q.push(t->left);
if (t->right) q.push(t->right);
}
return res;
}
};
扩展题:分行从上往下打印二叉树
总结
- 模拟顺时针打印矩阵
- 模拟栈的压入、弹出
- 层序遍历的模拟——BFS(宽度优先搜索)