劍指offer:給定一棵二叉搜尋樹,請找出其中的第k小的結點。

gaozhuang63發表於2020-11-10

劍指offer演算法題


題目描述
給定一棵二叉搜尋樹,請找出其中的第k小的結點。

題目分析
由於給定的是一顆二叉搜尋樹,而對於二叉搜尋樹來說,其中序遍歷的結果就是該樹升序排序後的結果。所以我們可以對該樹進行中序遍歷,然後找到第k個即可。

方法一 遞迴
下面是Java程式碼

/*
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}
*/
public class Solution {
	//設定兩個全域性變數,一個用於計數,一個用於儲存找到的節點。
    private int i =0;
    private TreeNode node = null;
    TreeNode KthNode(TreeNode pRoot, int k)
    {
        if(pRoot == null){
            return null;
        }
        if(k <=0){
            return null;
        }
        
        KthNode(pRoot.left,k);
        i++;
        if(i == k){
            node = pRoot;
        }
        KthNode(pRoot.right,k);
        
        return node;
    }
}

方法二 非遞迴

下面是Java程式碼

/*
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}
*/
import java.util.Stack;

public class Solution {
    TreeNode KthNode(TreeNode pRoot, int k)
    {
        if(pRoot == null){
            return null;
        }
        if(k <=0){
            return null;
        }
        
        Stack<TreeNode> s = new Stack<>();
        int count = 0;
        while(!s.isEmpty()||pRoot!=null){
            if(pRoot!=null){
                s.push(pRoot); //直接壓棧
                pRoot = pRoot.left;
            }else{
                TreeNode node = s.pop(); //出棧並訪問
                if(++count == k){
                    return node;
                }
                pRoot = node.right;
            }
            
        }
        
        return null;
    }
}

參考https://www.nowcoder.com/practice/ef068f602dde4d28aab2b210e859150a?tpId=13&&tqId=11215&rp=1&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking

相關文章