leetcode剑指offer32-III 广度优先算法

it2026-01-28  7

题目:https://leetcode-cn.com/problems/cong-shang-dao-xia-da-yin-er-cha-shu-iii-lcof/ 思路:广度优先算法 利用栈(先进后出)

定义两个栈,一个用于存储当前节点,一个用于存储下一层的节点判断根节点是否为零,为零就将返回空;将根节点入栈1,将标识符设置为1,(表示对栈1 内的节点进行操作)栈1、栈2不为空 执行循环体如果标识符为1,计算栈1节点的个数,将每个节点从栈中抛出,每抛出一个节点,将结点的值入一维数组,将节点的左子树入栈2,将节点的右子树入栈2,(左右子树入栈的顺序有要求),遍历完栈1中的节点后,将一维数组入二维数组,并将一维数组清空,将标识符设置为2;如果标识符为2,计算栈2点的个数,将每个节点从栈中抛出,每抛出一个节点,将结点的值入一维数组,将节点的左子树入栈1,将节点的右子树入栈1,(左右子树入栈的顺序有要求),遍历完栈2中的节点后,将一维数组入二维数组,并将一维数组清空,将标识符设置为1;直到所有的节点都遍历(栈1和栈2都为空); class Solution { public: vector<vector<int>> levelOrder(TreeNode* root) { vector<vector<int>> vec; stack<TreeNode*>stk1,stk2; int flag = 1; if (root == NULL) return vec; stk1.push(root); vector<int>A; while (!stk1.empty()|| !stk2.empty()) { if (flag == 1) { int len = stk1.size(); for (int i = 0; i < len; i++) { TreeNode*node = stk1.top(); stk1.pop(); A.push_back(node->val); if (node->left) stk2.push(node->left); if (node->right) stk2.push(node->right); } vec.push_back(A); A.clear(); flag = 2; } else { int len = stk2.size(); for (int i = 0; i < len; i++) { TreeNode*node = stk2.top(); stk2.pop(); A.push_back(node->val); if (node->right)stk1.push(node->right); if (node->left) stk1.push(node->left); } vec.push_back(A); A.clear(); flag = 1; } } return vec; } };
最新回复(0)