IT公司100题-tencent-打印所有高度为2的路径

问题描述：

打印所有到叶子节点长度为2的路径

　　  10

　　 /  \

　　6   16

　 / \   / \

　4 8  14 18

  / \    / \    \

2  5  12 15 20

　　 /

　 11

 

打印：

[10 6 8]

[6 4 2]

[6 4 5]

[16 14 15]

[16 18 20]

[14 12 11]

 

分析：

1， 对于树先做先序遍历。

2， 然后针对每个节点做检查，检查的内容是，是否含有长度为2的叶子节点路径。

3， 检查的过程是，设定一个depth，同时保存当前root节点，递归遍历每个子节点，每次递归depth+1，push当前root节点，如果depth为n，当前节点为叶子节点，则打印出来，否则如果depth<n，就继续遍历子节点，跟随每次入栈操作的是:1, depth+1 2, push_back(root)。如果depth>=n，就不继续遍历，同时跟随函数调用结束出栈的操作是：1，depth-1 2，pop当前节点。

 

代码实现：

 #include <iostream>
 #include "../utility/printCollection.h"
 using namespace std;

 struct bsnode{
     int data;
     bsnode* left;
     bsnode* right;
 };

 void add_bsnode(bsnode* &root, int value)
 {
     if(root == NULL)
     {
         bsnode* tmp = new bsnode();
         tmp->data = value;
         tmp->left = NULL;
         tmp->right = NULL;
         root = tmp;
         return;
     }
     if(value > root->data)
         add_bsnode(root->right, value);
     else if(value < root->data)
         add_bsnode(root->left, value);
     else
         cout << "duplicate value" << endl;
 }

 void print_tree(bsnode* root)
 {
     if(root == NULL)
         return;
     if(root->left != NULL)
         print_tree(root->left);
     cout << root->data << " ";
     if(root->right != NULL)
         print_tree(root->right);
 }

 void check_node(bsnode* root, int n, int &depth, vector<int> &path)
 {
     //如果当前深度>n，不需要继续访问
     if(depth > n || root == NULL)
         return;

     //检查前压栈当前节点，depth+1
     path.push_back(root->data);
     depth++;

     //如果当前深度=n
     if(depth == n)
         if(root->left == NULL && root->right == NULL)
         {
             cout << "path found: " << endl;
             printCollection(path);
             cout << endl;
         }

     //如果当前深度<n
     if(depth < n)
         if(root->left != NULL)
             check_node(root->left, n, depth, path);
         if(root->right != NULL)
             check_node(root->right, n, depth, path);

     //检查完出栈当前节点，depth-1
     path.pop_back();
     depth--;
     return;
 }

 void print_node(bsnode* root, int n)
 {
     if(root == NULL)
         return;

     //每次检查前重定义depth
     int depth = -;
     vector<int> path;

     //每次检查前clear path
     path.clear();

     //检查当前节点的所有叶子节点的高度
     check_node(root, n, depth, path);

     //递归检查左子树
     print_node(root->left, n);
     //递归检查右子树
     print_node(root->right, n);
 }

 int main()
 {
     bsnode* root = NULL;
     add_bsnode(root, );
     add_bsnode(root, );
     add_bsnode(root, );
     add_bsnode(root, );
     add_bsnode(root, );
     add_bsnode(root, );
     add_bsnode(root, );
     add_bsnode(root, );
     add_bsnode(root, );
     add_bsnode(root, );
     add_bsnode(root, );
     add_bsnode(root, );
     add_bsnode(root, );

     cout << "Tree content: " << endl;
     print_tree(root);
     cout << endl << endl;

     int height = ;
     cout << "Nodes that with height " << height << " are:" << endl;
     print_node(root, height);
     cout << endl;
 }

输出：

$ ./a.exe
Tree content:

Nodes that with height  are:
path found:
[  ]

path found:
[  ]

path found:
[  ]

path found:
[  ]

path found:
[  ]

path found:
[  ]