DFS有递归与非递归两种常见形式,BFS则通常为非递归的
本文使用TreeNode.h
如下
struct TreeNode {
int val;
TreeNode* left;
TreeNode* right;
TreeNode() : val(0), left(nullptr), right(nullptr) {}
TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
TreeNode(int x, TreeNode* left, TreeNode* right) : val(x), left(left), right(right) {}
};
1. DFS
递归法
以前序递归法为例
c++
class Solution {
public:
void preorder(TreeNode* root, vector<int> res){
if (!root) return;
res.push_back(root->val);
preorder(root->left, res);
preorder(root->right, res);
}
vector<int> preorderTraversal(TreeNode* root) {
vector<int> res;
preorder(root, res);
return res;
}
};
中序后序递归只需改变preorder
函数的语句顺序即可
非递归形式
仍以前序DFS为例
c++
class Solution {
public:
vector<int> preorderTraversal(TreeNode* root) {
stack<TreeNode*> st;
vector<int> result;
st.push(root);
while (!st.empty()) {
TreeNode* node = st.top(); // 中
st.pop();
if (node != NULL) result.push_back(node->val);
else continue;
st.push(node->right); // 右
st.push(node->left); // 左
}
return result;
}
};
2. BFS
c++
#include<TreeNode.h>
#include<vector>
#include<queue>
using namespace std;
class Solution {
public:
vector<int> bfs(TreeNode* root) {
vector<int> res;
queue<TreeNode*> que;
que.push(root);
while (!que.empty()) {
TreeNode* tmpNode = que.front();
que.pop();
if (tmpNode->left != nullptr) {
que.push(tmpNode->left);
res.push_back(tmpNode->left->val);
}
if (tmpNode->right != nullptr) {
que.push(tmpNode->right);
res.push_back(tmpNode->right->val);
}
}
return res;
}
};
这是比较标准的bfs形式,应作为一个基本类型记忆。结合下面图中的这个例子比较好记,
- 循环前初始化根结点进队列
- 循环开始出队一个元素
- 其后使左右孩子都进入队列
- 队列不为空时循环往复
合理性可以画图进行分析
start代表循坏之前初始化的队列,圆圈中元素是出队列元素\当前结点
。队列右进左出
层序遍历
在BFS的基础上记录队列长度,可以进行leetcode-102层序遍历
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
queue<TreeNode*> que;
if (root != NULL) que.push(root);
vector<vector<int>> res;
while (!que.empty()) {
int size = que.size();
vector<int> vec; //准备工作
for(int i = 0; i < size; i++){ //内层才是明显的bfs
TreeNode* node = que.front();
que.pop();
vec.push_back(node->val);
if (node->left) que.push(node->left);
if (node->right) que.push(node->right);
}
res.push_back(vec);
}
return res;
}
};