1. 程式人生 > >輸入一個整數和一棵二元樹,從樹的根結點開始往下訪問一直到葉結點所經過的所有結點形成一條路徑,打印出和與輸入整數相等的所有路徑

輸入一個整數和一棵二元樹,從樹的根結點開始往下訪問一直到葉結點所經過的所有結點形成一條路徑,打印出和與輸入整數相等的所有路徑

題目:輸入一個整數和一棵二元樹。從樹的根結點開始往下訪問一直到葉結點所經過的所有結點形成一條路徑。打印出和與輸入整數相等的所有路徑。例如 輸入整數22和如下二元樹
        10   
      /    \   
     5       12   
   /   \   
  4     7
則打印出兩條路徑:10, 12和10, 5, 7。
二元樹節點的資料結構定義為:
struct BinaryTreeNode // a node in the binary tree
{
int m_nValue; // value of node
BinaryTreeNode *m_pLeft; // left child of node
BinaryTreeNode *m_pRight; // right child of node

};

看到這個題目,首先想到的是,先找到滿足的葉子。但是題目要求是列印從跟節點到葉子的路徑,所以還得把路徑打出來。

我首先想到的是使用vector容器(程式碼裡面包含的有構造一個最簡單的二叉查詢樹,新增節點,列印整棵樹(用中序遍歷))

<span style="color:#009900;">#include <iostream>
#include <algorithm>
using namespace std;

class CTwoTree
{
public:
    struct BSTreeNode
    {
      int m_nValue; // value of node
      BSTreeNode *m_pLeft; // left child of node
      BSTreeNode *m_pRight; // right child of node
    };
    CTwoTree(){proot = NULL;}
    void CAddNode(int m_nValue, BSTreeNode * &p)
    {
        if(p == NULL)
        {
            p = new BSTreeNode;
            p->m_nValue = m_nValue;
            p->m_pLeft = NULL;
            p->m_pRight = NULL;
            return;
        }
        if(m_nValue > p->m_nValue)
        {
            CAddNode(m_nValue, p->m_pRight);
        }
        else if(p->m_nValue > m_nValue)
        {
            CAddNode(m_nValue, p->m_pLeft);
        }

    }
    void CAddNode(int m_nValue)
    {
        CAddNode(m_nValue, proot);
    }
    void zhongxu(BSTreeNode *root)
    {
     if(root == NULL)
        return;
     if(root->m_pLeft != NULL)
     {
        zhongxu(root->m_pLeft);
     }
    cout << root->m_nValue << endl;
     if(root->m_pRight != NULL)
     {
        zhongxu(root->m_pRight);
     }
    }
    void printTree()
    {
        zhongxu(proot);
    }
    void findSum(int x, BSTreeNode *&p)
    {
        if(p == NULL)
            return;
        path.push_back(p);//每次進來先存起來,不滿足則pop掉
        if(p->m_pLeft != NULL)
        {
            findSum(x-p->m_nValue,p->m_pLeft);
        }
        if(p->m_pRight != NULL)
        {
            findSum(x-p->m_nValue,p->m_pRight);
        }
        if(x == p->m_nValue && p->m_pLeft == NULL && p->m_pRight == NULL)
        {
            /********為了列印成逗號間隔的格式**********/
            for(int i = 0; i < path.size()-1; i ++)
                cout << path.at(i)-> m_nValue<< ",";
            cout  << path.at(path.size()-1)-> m_nValue << endl;

            /********為了返回到上一節點**********/
            while(1)
            {
                if(path.size() >= 1)
                {
                    if(path.at(path.size()-1)->m_pRight == NULL)
                        path.pop_back();
                    else
                        break;
                }
                else
                    break;
            }
        }
        else
            path.pop_back();//不滿足則pop掉

    }
    void findSum(int x)
    {
        findSum(x, proot);
    }
    void deleteAll(BSTreeNode * p)
    {
<span style="white-space:pre">	</span>delete p;
    }
    ~CTwoTree()
    {<span style="white-space:pre">
</span><span style="white-space:pre">	</span>for_each(path.begin(),path.end(),deleteAll);

    }
private:
    BSTreeNode *proot;
    vector<BSTreeNode*> path;
};
int main(int argc, char *argv[])
{
    CTwoTree a;
    a.CAddNode(10);
    a.CAddNode(6);
    a.CAddNode(4);
    a.CAddNode(8);
    a.CAddNode(14);
    a.CAddNode(12);
    a.CAddNode(16);

    a.printTree();
    a.findSum(40);

    return 0;
}

</span>