我想了两种方法,一种就是将二叉树进行前序遍历,并将值全部扔进一个容器里面,这样容器里面的值就是递增排序的数,输出第k-1个值 ,就ok,这个方法很容易理解。
第二个就是用栈的性质,将树的左值先扔到栈里面,然后再弹栈,每次弹栈都将此节点的右值扔进栈里,这样就能进行递增遍历,加上k–,当k=0时正好输出就ok。
struct TreeNode {
int val;
TreeNode *left;
TreeNode *right;
TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};
class Solution {
public:
void pushvor(vector&cc, TreeNode* root)
{
while (root == NULL)return;
pushvor(cc, root->left);
cc.push_back(root->val);
pushvor(cc, root->right);
return;
}
int kthSmallest(TreeNode* root, int k)
{
vector<int>cc;
pushvor(cc, root);
return cc[k - 1];
}
};
/**
-
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:int kthSmallest(TreeNode* root, int k)
{
stack<TreeNode*>cc;
while(root||!cc.empty())
{
while(root)
{
cc.push(root);
root=root->left;
}root=cc.top(); cc.pop(); k--; if(k==0)return root->val; if(root->right)root=root->right; else root=NULL; } return 0;
}
};