題目地址:
https://leetcode-cn.com/probl...
題目描述:
給定一個二叉搜索樹,編寫一個函數 kthSmallest 來查找其中第 k 個最小的元素。
說明:
你可以假設 k 總是有效的,1 ≤ k ≤ 二叉搜索樹元素個數。
示例 1:
輸入: root = [3,1,4,null,2], k = 1
3
/ \
1 4
\
2
輸出: 1
示例 2:
輸入: root = [5,3,6,2,4,null,null,1], k = 3
5
/ \
3 6
/ \
2 4
/
1
輸出: 3
解答:利用二叉搜索樹的中序遍歷是有序的特性。
可以中序遍歷的時候用一個數組去存,然後取出數組的第k個數。不過這樣似乎有些浪費空間。
因此優化的方法是,可以在中序遍歷的時候記錄每次遍歷到第幾個節點。
如果該節點正是第k個節點,那麼記錄。並且停止之後所有的遍歷。
java ac代碼:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
//記錄答案
int ans;
//表示當前節點的位置
int pos;
public int kthSmallest(TreeNode root, int k) {
dfs(root,k);
return ans;
}
void dfs(TreeNode root,int k)
{
if(root == null)return;
dfs(root.left,k);
//這個剪枝,否則仍是遍歷每一個節點。
//pos等於k表示已經求出了第k個節點。
if(pos == k)return;
pos++;
if(pos == k)
{
ans = root.val;
return;
}
dfs(root.right,k);
}
}