/*求二叉樹深度 -- 採用遞迴和非遞迴方法
**經除錯可執行原始碼及分析如下：
*/
#include <stdio.h>
#include <stdlib.h>
#include <iostream>
#include <queue>

using namespace std;
/*二叉樹結點定義*/
typedef struct BTreeNode
{
    char elem;
    struct BTreeNode *pleft;
    struct BTreeNode *pright;
}BTreeNode;

/*
*如果根節點為NULL，則深度為0
*如果根節點不為NULL，則深度 = 左右子樹的深度的最大值+1
*/
 


/*遞迴實現求深度*/
int get_depth(BTreeNode *proot)
{
    if (proot == NULL)
    {
        return 0;
    }

    int left_depth = get_depth(proot->pleft);
    int right_depth = get_depth(proot->pright);

    return (((left_depth > right_depth) ? left_depth:right_depth) + 1);
}
/**********************************************************/
 

/*非遞迴：藉助佇列，在進行按層遍歷時，記錄遍歷的層數即可。
 **/
/*非遞迴實現求深度*/
int get_depth2(BTreeNode* proot)
{
    int depth = 0;
    if (proot == NULL)
    {
        return 0;
    }

    queue <BTreeNode *> que;
    que.push(proot);
    while (!que.empty())
    {
        ++depth;
        int cur_level_nodes_count = que.size();//當前層節點數量
 

        int temp_count = 0;//計數器：當前層次節點個數
        while (temp_count < cur_level_nodes_count)
        {
            ++temp_count;
            proot = que.front();
            que.pop();
            if (proot->pleft != NULL)
            {
                que.push(proot->pleft);
            }
            if (proot->pright != NULL)
            {
                que.push(proot->pright);
            }
        }
    }
    return depth;
}
/**********************************************************/

/*初始化二叉樹根節點*/
BTreeNode* btree_init(BTreeNode* &bt)
{
    bt = NULL;
    return bt;
}

/*先序建立二叉樹*/
void pre_crt_tree(BTreeNode* &bt)
{
    char ch;
    cin >> ch;
    if (ch == '#')
    {
        bt = NULL;
    }
    else
    {
        bt = new BTreeNode;
        bt->elem = ch;
        pre_crt_tree(bt->pleft);
        pre_crt_tree(bt->pright);
    }
}

int main()
{
    int tree_depth = 0;
    BTreeNode *bt;
    btree_init(bt);//初始化根節點
    pre_crt_tree(bt);//建立二叉樹
    tree_depth = get_depth(bt);//遞迴
    cout << "二叉樹深度為:" << tree_depth << endl;
    tree_depth = get_depth2(bt);//非遞迴
    cout << "二叉樹深度為:" << tree_depth << endl;

    system("pause");
    return 0;
}


/*
執行結果：

a b c # # # d # #

---以上為輸入---
---以下為輸出---

二叉樹深度為:3
二叉樹深度為:3
請按任意鍵繼續. . .


本例建立的二叉樹形狀：
        a
    b       d   
c

參考資料：
http://blog.csdn.net/beitiandijun/article/details/41930583
http://yuncode.net/code/c_505ea04f8f6186
*/