1、先序遍历
首先访问根节点,然后根据栈后进先出的特点,先访问右节点,再左节点。这样出栈的时候就是正确的结果。
void iterativePreorder(Node *node)
{
if (node == nullptr) {
return;
}
stack<Node*> stack;
stack.push(node);
while (!stack.empty()) {
node = stack.top();
stack.pop();
// 先访问节点
visit(node);
// 把右子结点压入栈
if (node.right != null) {
stack.push(node.right);
}
// 把左子结点压入栈
if (node.left != null) {
stack.push(node.left);
}
}
}
2、中序遍历
首先访问左节点,然后根据栈后进先出的特点,先访问右节点,再根节点。
void iterativeInOrder(Node *root)
{
if (root == nullptr) {
return;
}
stack<Node*> stack;
Node *node = root;
while (node != nullptr || !stack.empty()) {
// 把当前节点的所有左侧子结点压入栈
while (node != nullptr) {
stack.push(node); // 存储的是根,左, 根,左
node = node->left;
}
// 访问节点,处理该节点的右子树
if (!stack.empty()) {
node = stack.top();
stack.pop();
visit(node); // 访问左,并退回到根。然后处理右,那么栈中就是右,根,下次访问这里的时候顺序就会是根,右
node = node->right;
}
}
}
3、后序遍历
void iterativePostOrderByTwoStacks(Node *root)
{
if (root == nullptr) {
return;
}
stack<Node*> stack;
stack<Node*> tmp;
Node *node = root;
while (node != nullptr || !stack.empty()) {
// 把当前节点和其右侧子结点推入栈
while (node != nullptr) {
stack.push(node);
tmp.push(node); // tmp中存储的就是根,右,(左),根,右,(左),...
node = node->riqht;
}
// 处理栈顶节点的左子树
if (!stack.empty()) {
node = stack.top();
stack.pop();
node = node->left; // 获取到left节点,left就在栈顶
}
}
// 存储的是根,右,左,出来的就是左,右,根
while (!tmp.empty()) {
node = tmp.top();
tmp.pop();
visit(node);
}
}
4、层序遍历
层序用队列实现
void layerTraversal(Node *root)
{
if (root == null) {
return;
}
queue<Node*> queue();
queue.push(root);
while (!queue.isEmpty()) {
Node *currentNode = queue.front();
queue.pop();
visit(currentNode);
if (currentNode.left != null) {
queue.push(currentNode.left);
}
if (currentNode.right != null) {
queue.push(currentNode.right);
}
}
}