深度優先搜索
- 思路：
- 二叉搜索樹的特性，通過中序遍歷得到有序序列，則遍歷到第K個節點的時候即為結果；
- 使用棧通過深度優先遍歷進行中序遍歷：
- 先將節點和左子節點壓棧；
- 然后棧頂上就是“最左”葉子節點；
- 然后通過棧回溯其父節點 p ，將 p 右子樹壓棧，重復上述過程；
/*** Definition for a binary tree node.* struct TreeNode {* int val;* TreeNode *left;* TreeNode *right;* TreeNode() : val(0), left(nullptr), right(nullptr) {}* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}* };*/
class Solution {
public:int kthSmallest(TreeNode* root, int k) {std::stack<TreeNode*> stack;TreeNode* node = root;while (node != nullptr || stack.size() > 0) {while (node != nullptr) {stack.push(node);node = node->left;}node = stack.top();stack.pop();--k;if (k == 0) {break;}node = node->right;}return node->val;}
};
