二叉树的问题,第一想到递归,其次想到各种遍历方式的变种。
一、二叉树的定义
struct TreeNode {
int val;
TreeNode *left;
TreeNode *right;
TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};
二、前序遍历(根左右)
利用栈,先往左下走,并一路进栈,并访问当前节点。
若当前节点为空,就回到栈顶节点,往右下走一个。继续往左下走。
void preorderTraversal(TreeNode* root) {
stack<TreeNode*> stk;
TreeNode* cur = root;
while(cur || !stk.empty()){
while(cur){
stk.push(cur);
cout<<cur->val;
cur=cur->left;
}
cur=stk.top();
stk.pop();
cur=cur->right;
}
}
三、中序遍历(左根右)
与前序遍历类似,但是从左下返回时才访问当前节点。
void inorderTraversal(TreeNode* root) {
stack<TreeNode*> stack;
TreeNode* cur = root;
while(cur || !stack.empty()){
while(cur){
stack.push(cur);
cur=cur->left;
}
cur = stack.top();
stack.pop();
cout<<cur->val;//访问时机是前序与中序的唯一不同
cur=cur->right;
}
}
四、后序遍历(左右根)
后序应该是最难的。
整体框架与前序中序一样,但在访问当前节点时要判断右下节点是否为空或是上一个访问的节点。
void postorderTraversal(TreeNode* root) {
TreeNode* lastVisited = nullptr;
stack<TreeNode*> stk;
TreeNode* cur = root;
while(cur || !stk.empty()){
while(cur){
stk.push(cur);
cur=cur->left;
}
cur=stk.top();
//当右子节点为空,或者右子节点为上一个访问的节点,才访问当前节点
if(!cur->right || cur->right==lastVisited){
cout<<cur->val;
lastVisited=cur;
stk.pop();
cur=nullptr;//重要,如果不设置为空,则cur还会继续进栈
}
else{
cur=cur->right;
}
}
}
五、层次遍历(每层一行)
利用队列。队首节点出栈并访问,则其左右非空节点入队。
如果只需要打印一行则更加简单。
void levelOrder(TreeNode* root) {
queue<TreeNode*> que;
que.push(root);
while(!que.empty()){
int size = que.size();
for(int i=0; i<size; i++){
TreeNode* cur = que.front();
que.pop();
cout<<cur->val;
if(cur->left) que.push(cur->left);
if(cur->right) que.push(cur->right);
}
cout<<endl;
}
}
六、递归算法
比较简单。
vector<int> res;
vector<int> postorderTraversal(TreeNode* root) {
if(root==NULL)
return res;
//res.push_back(root->val);写在此处就是前序遍历
postorderTraversal(root->left);
//res.push_back(root->val);写在此处就是中序遍历
postorderTraversal(root->right);
res.push_back(root->val);//写在此处就是后序遍历
return res;
}