一、題目：二叉搜尋樹的後序遍歷序列

題目：輸入一個整數陣列，判斷該陣列是不是某二叉搜尋樹的後序遍歷的結果。如果是則返回true，否則返回false。假設輸入的陣列的任意兩個數字都互不相同。

　　例如在下面的一顆二叉搜尋樹中，輸入陣列{5,7,6,9,11,10,8}，則返回true，因為這個整數序列是下圖二叉搜尋樹的後序遍歷結果。如果輸入的陣列是{7,4,6,5}，由於沒有哪棵二叉搜尋樹的後序遍歷的結果是這個序列，因此返回false。

二、解題思路

2.1 核心步驟

　　在後序遍歷得到的序列中，最後一個數字是樹的根結點的值。陣列中前面的數字可以分為兩部分：第一部分是左子樹結點的值，它們都比根結點的值小；第二部分是右子樹結點的值，它們都比根結點的值大

　　因此，我們可以總結出演算法步驟：

　　Step1.通過取出序列最後一個元素得到二叉搜尋樹的根節點；

　　Step2.在二叉搜尋樹中左子樹的結點小於根結點，因此可以遍歷一次得到左子樹；

　　Step3.在二叉搜尋樹中右子樹的結點大於根結點，因此可以繼續遍歷後序元素得到右子樹；

　　Step4.重複以上步驟遞迴判斷左右子樹是不是二叉搜尋樹，如果都是，則返回true，如果不是，則返回false;

2.2 程式碼實現

    public static bool VerifySquenceOfBST(int[] sequence, int length)
    {
        
 
if (sequence == null || length <= 0)
        {
            return false;
        }

        int root = sequence[length - 1];

        int i = 0;
        // 在二叉搜尋樹中左子樹的結點小於根結點
        for (; i < length - 1; i++)
        {
            if (sequence[i] > root)
            {
                
 
break;
            }
        }
        // 在二叉搜尋樹中右子樹的結點大於根結點
        int j = i;
        for (; j < length - 1; j++)
        {
            if (sequence[j] < root)
            {
                // 如果找到小於根節點直接返回false
                return false;
            }
        }
        // 判斷左子樹是不是二叉搜尋樹
        bool leftIsBST = true;
        if (i > 0)
        {
            leftIsBST = VerifySquenceOfBST(sequence, i);
        }
        // 判斷右子樹是不是二叉搜尋樹
        bool rightIsBST = true;
        if (j < length - 1)
        {
            // C#中無法直接操作指標，在C/C++可以直接傳遞sequence+i
            int[] newSequence = sequence.Skip(i).ToArray();
            rightIsBST = VerifySquenceOfBST(newSequence, length - i - 1);
        }

        return leftIsBST && rightIsBST;
    }

三、單元測試

3.1 測試用例

    //            10
    //         /      \
    //        6        14
    //       /\        /\
    //      4  8     12  16
    [TestMethod]
    public void SequenceTest1()
    {
        int[] data = { 4, 8, 6, 12, 16, 14, 10 };
        bool result = SequenceHelper.VerifySquenceOfBST(data, data.Length);
        Assert.AreEqual(result, true);
    }

    //           5
    //          / \
    //         4   7
    //            /
    //           6
    [TestMethod]
    public void SequenceTest2()
    {
        int[] data = { 4, 6, 7, 5 };
        bool result = SequenceHelper.VerifySquenceOfBST(data, data.Length);
        Assert.AreEqual(result, true);
    }

    //               5
    //              /
    //             4
    //            /
    //           3
    //          /
    //         2
    //        /
    //       1
    [TestMethod]
    public void SequenceTest3()
    {
        int[] data = { 1, 2, 3, 4, 5 };
        bool result = SequenceHelper.VerifySquenceOfBST(data, data.Length);
        Assert.AreEqual(result, true);
    }

    // 1
    //  \
    //   2
    //    \
    //     3
    //      \
    //       4
    //        \
    //         5
    [TestMethod]
    public void SequenceTest4()
    {
        int[] data = { 5, 4, 3, 2, 1 };
        bool result = SequenceHelper.VerifySquenceOfBST(data, data.Length);
        Assert.AreEqual(result, true);
    }

    // 樹中只有1個結點
    [TestMethod]
    public void SequenceTest5()
    {
        int[] data = { 5 };
        bool result = SequenceHelper.VerifySquenceOfBST(data, data.Length);
        Assert.AreEqual(result, true);
    }

    // 錯誤序列
    [TestMethod]
    public void SequenceTest6()
    {
        int[] data = { 7, 4, 6, 5 };
        bool result = SequenceHelper.VerifySquenceOfBST(data, data.Length);
        Assert.AreEqual(result, false);
    }

    // 錯誤序列
    [TestMethod]
    public void SequenceTest7()
    {
        int[] data = { 4, 6, 12, 8, 16, 14, 10 };
        bool result = SequenceHelper.VerifySquenceOfBST(data, data.Length);
        Assert.AreEqual(result, false);
    }

    // 錯誤序列
    [TestMethod]
    public void SequenceTest8()
    {
        bool result = SequenceHelper.VerifySquenceOfBST(null, 0);
        Assert.AreEqual(result, false);
    }

3.2 測試結果

作者：周旭龍

本文版權歸作者和部落格園共有，歡迎轉載，但未經作者同意必須保留此段宣告，且在文章頁面明顯位置給出原文連結。