[关闭]
@kakadee 2016-07-14T02:17:19.000000Z 字数 3125 阅读 1840

二叉树的常见算法1


1. 二叉树的存储结构

  1. struct TreeNode
  2. {
  3. int val;
  4. TreeNode *left;
  5. TreeNode *right;
  6. TreeNode(int x) : val(x), left(NULL), right(NULL) {}
  7. };

2. 根据字符串创建一颗二叉树

1. 描述

给你一串字符串,每个节点用‘,’隔开,用‘#’代表空的字符串。按从上到下的顺序给出字符串。
例如:
字符串---1,2,#,3,4 对应的二叉树为:
image_1anfl77339s9v6817pt10nngiu9.png-6.2kB

2. 思路

  1. 根据‘,’的位置分离出每一个节点,存储在vector数组中。
  2. 建立每一个节点的指针关系。
    image_1anfmjegirv2eq8htocavh4dm.png-7.6kB

3. 代码

  1. TreeNode* Deserialize(string s) {
  2. if (s.size() == 0) return NULL;
  3. vector<TreeNode*> path; // 用来存储每个树节点
  4. while (s != "") {
  5. int i = 0;
  6. while (i < s.size() && s[i] != ',') i++;
  7. string tmp = s.substr(0, i); // 分离出一个节点
  8. if (tmp == "#") {
  9. TreeNode* p = NULL;
  10. path.push_back(p);
  11.                         
  12. }
  13. else {
  14. int val = atoi(tmp.c_str()); // 将字符串转换成整型
  15. TreeNode* p = new TreeNode(val);
  16. path.push_back(p);
  17.                         
  18. }
  19. if (i >= s.size() - 1) s = "";
  20. else s = s.substr(i + 1);                 
  21. }
  22. int index = 1, i = 0;
  23. // index记录的是i的子树的位置
  24. while (index < path.size()) {
  25. if (path[i] != NULL) {
  26. if (index < path.size()) path[i]->left = path[index++];
  27. if (index < path.size()) path[i]->right = path[index++];             
  28. }
  29. i++;    
  30. }
  31. return path[0];     
  32. }

3. 将二叉树还原成字符串

描述

题2的镜像问题,给你一颗二叉树,将其还原成字符串。
这里我是按层次遍历的方法写的,如果有疑问,可以转到层次遍历二叉树

  1. string Serialize(TreeNode *root) {
  2. string ans;
  3. if (root == NULL) return &ans[0];
  4. queue<TreeNode *> Q;
  5. Q.push(root);
  6. while (!Q.empty()) {
  7. TreeNode *node = Q.front();
  8. Q.pop();
  9. if (node == NULL) ans += ",#";
  10. else {
  11. Q.push(node->left);
  12. Q.push(node->right);
  13. string tmp = to_string(node->val);
  14. ans += "," + tmp;       
  15. }
  16. }
  17. int i = ans.size()-1;
  18. while(i >= 0) { // 去除字符串末尾的#字符
  19. if (ans[i] == '#' || ans[i] == ',')
  20. i--;
  21. else {
  22. break;
  23. }
  24. }
  25. return ans.substr(1,i);     
  26. }

4.二叉树的前序遍历(递归+非递归)

1. 递归方法

前序遍历的访问顺序是 根 -> 左 -> 右

  1. void preorder(TreeNode *root, vector<int> &ans) {
  2. if (root != NULL) {
  3. ans.push_back(root->val);
  4. preorder(root->left,ans);
  5. preorder(root->right,ans);
  6. }
  7. }

2. 非递归方法

利用栈的FIFO性质,先访问栈的顶层节点,然后先把当前节点的右子树入栈,再把当前节点的左子树入栈。当栈不为空的时候继续访问栈。

  1. void preorder2(TreeNode *root, vector<int> &ans) {
  2. if (root == NULL) return;
  3. stack<TreeNode *> S;
  4. S.push(root);
  5. // 利用栈的FIFO性质,先将节点的右子树入栈,再将节点的左子树入栈
  6. while(!S.empty()) {
  7. TreeNode *node = S.top();
  8. S.pop();
  9. ans.push_back(node->val);
  10. if (node->right) S.push(node->right);
  11. if (node->left) S.push(node->left);
  12. }
  13. }

5. 二叉树的中序遍历

1. 递归方法

中序遍历的访问顺序: 左 -> 根 -> 右

  1. void inorder(TreeNode *root,vector<int> &ans) {
  2. if (root != NULL) {
  3. inorder(root->left,ans);
  4. ans.push_back(root->val);
  5. inorder(root->right, ans);
  6. }
  7. }

2. 非递归方法

对于任一结点,优先访问其左孩子,而左孩子结点又可以看做一根结点,然后继续访问其左孩子结点,直到遇到左孩子结点为空的结点才进行访问,然后按相同的规则访问其右子树。

 对于任一结点P,

  1. 若其左孩子不为空,则将P入栈并将P的左孩子置为当前的P,然后对当前结点P再进行相同的处理;
  2. 其左孩子为空,则取栈顶元素并进行出栈操作,访问该栈顶结点,然后将当前的P置为栈顶结点的右孩子;
  3. 直到P为NULL并且栈为空则遍历结束
  1. void inorder2(TreeNode *root,vector<int> &ans) {
  2. if (root == NULL) return;
  3. stack<TreeNode *> S;
  4. TreeNode *p = root;
  5. while(p != NULL || !S.empty()) {
  6. while (p) {
  7. S.push(p);
  8. p = p->left;
  9. }
  10. if (!S.empty()) {
  11. p = S.top();
  12. S.pop();
  13. ans.push_back(p->val);
  14. p = p->right;
  15. }
  16. }
  17. }

6. 二叉树的后序遍历

1. 递归方法

后序遍历的访问顺序: 左 -> 右 -> 根

  1. void postorder(TreeNode *root,vector<int> &ans) {
  2. if (root != NULL) {
  3. postorder(root->left,ans);
  4. postorder(root->right, ans);
  5. ans.push_back(root->val);
  6. }
  7. }

2. 非递归方法

要保证根结点在左孩子和右孩子访问之后才能访问,因此对于任一结点P,先将其入栈。如果P不存在左孩子和右孩子,则可以直接访问它;或者P存在左孩子或者右孩子,但是其左孩子和右孩子都已被访问过了,则同样可以直接访问该结点。若非上述两种情况,则将P的右孩子和左孩子依次入栈,这样就保证了每次取栈顶元素的时候,左孩子在右孩子前面被访问,左孩子和右孩子都在根结点前面被访问。
参考海子的博客:博客地址

  1. void postorder2(TreeNode *root,vector<int> &ans) {
  2. if (root == NULL) return;
  3. TreeNode *cur,*pre = NULL;
  4. stack<TreeNode *> S;
  5. S.push(root);
  6. while(!S.empty()) {
  7. cur = S.top();
  8. //如果当前结点没有孩子结点或者孩子节点都已被访问过
  9. if ( (!cur->left && !cur->right) || ( pre && (pre == cur->left || pre == cur->right))) {
  10. ans.push_back(cur->val);
  11. S.pop();
  12. pre = cur;
  13. }
  14. else { // 否则,右子树先入栈,接着左子树入栈
  15. if (cur->right) S.push(cur->right);
  16. if (cur->left) S.push(cur->left);
  17. }
  18. }
  19. }
添加新批注
在作者公开此批注前,只有你和作者可见。
回复批注