leetcode 230. Kth Smallest Element in a BST

最新推荐文章于 2024-10-01 05:04:32 发布

zbxzc

最新推荐文章于 2024-10-01 05:04:32 发布

阅读量450

点赞数

分类专栏：数据结构与算法 OJ 文章标签： leetcode

本文链接：https://blog.csdn.net/u014568921/article/details/50781841

版权

数据结构与算法同时被 2 个专栏收录

213 篇文章 1 订阅

订阅专栏

196 篇文章 0 订阅

订阅专栏

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 elements.

Follow up:

What if the BST is modified (insert/delete operations) often and you need to find the kth smallest frequently? How would you optimize the kthSmallest routine?

struct TreeNode {
	int val;
	TreeNode *left;
	TreeNode *right;
	TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};

class Solution {
private:
	int cur;
	int val;
	void traverse(TreeNode* node, int k)
	{
		if(node!=NULL)
		{
			if(node->left!=NULL)
			{
				traverse(node->left,k);
			}
			cur++;
			//cout<<node->val<<endl;
			if(cur==k)
			{
				val=node->val;
				return;
			}
			if(node->right!=NULL)
			{
				traverse(node->right,k);
			}
		}
	}
public:
	int kthSmallest(TreeNode* root, int k) {
		cur=0;
		val=-1;
		traverse(root,k);
		return val;
	}
};

这个很简单。左-根-右顺序遍历即可

accepted