230. Kth Smallest Element in a BST

1. 原题

Given a binary search tree, write a function kthSmallest to find the kth smallest element in it.

Note:
You may assume k is always valid, 1 ≤ k ≤ BST’s total element.

2. 分析

• 题目要求找出宽度优先树中的第K小的树。
• 在没有这棵树的数据的情况下不知道哪些数是存在的， 所以要确定第K个树， 必须要遍历一遍树。得到所有数之后排序就可以得到第K小的数了
• 宽度优先和深度优先的遍历都有规律可循， 分别判断根， 坐子树和右子树， 基本普通的遍历都是这个模式， 需要考虑的是退出情况还有传参的方式， 这里使用引用传参比较方便

 if (root != NULL){            nums.push_back(root->val);        }        if (root->left != NULL) getNums(root->left, nums);        if (root->right != NULL) getNums(root->right, nums);}

3.代码

/** * Definition for a binary tree node. * struct TreeNode { *     int val; *     TreeNode *left; *     TreeNode *right; *     TreeNode(int x) : val(x), left(NULL), right(NULL) {} * }; */class Solution {public:    void getNums(TreeNode* root, vector<int>& nums){        if (root != NULL){            nums.push_back(root->val);        }        if (root->left != NULL) getNums(root->left, nums);        if (root->right != NULL) getNums(root->right, nums);    }    int kthSmallest(TreeNode* root, int k) {        vector<int> nums;        int len = nums.size();        getNums(root, nums);        sort(nums.begin(), nums.end());        return nums[k-1];    }};