/
230_findkth_smallest.c
59 lines (31 loc) · 1.01 KB
/
230_findkth_smallest.c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
struct TreeInfo{
int numNodesVisited;
int lastVisitedValue;
} ;
void inOrderTraverse(struct TreeNode* root, int k , struct TreeInfo* treeInfo){
if (root == NULL || treeInfo->numNodesVisited >= k ){
return;
}
//process the left subtree
inOrderTraverse(root->left, k, treeInfo);
if (treeInfo -> numNodesVisited < k){
treeInfo -> numNodesVisited += 1;
treeInfo -> lastVisitedValue = root->val;
inOrderTraverse(root->right, k, treeInfo);
}
}
int kthSmallest(struct TreeNode* root, int k){
struct TreeInfo* treeInfo = (struct TreeInfo *)malloc(sizeof(struct TreeInfo));
treeInfo-> numNodesVisited = 0 ;
treeInfo-> lastVisitedValue = -1;
inOrderTraverse(root, k, treeInfo);
return treeInfo -> lastVisitedValue;
}