Marche dans l'ordre récursive avec un compteur
Time Complexity: O( N ), N is the number of nodes
Space Complexity: O( 1 ), excluding the function call stack
L'idée est similaire à la solution @prasadvk, mais elle présente quelques lacunes (voir les notes ci-dessous), je publie donc ceci comme une réponse séparée.
// Private Helper Macro
#define testAndReturn( k, counter, result ) \
do { if( (counter == k) && (result == -1) ) { \
result = pn->key_; \
return; \
} } while( 0 )
// Private Helper Function
static void findKthSmallest(
BstNode const * pn, int const k, int & counter, int & result ) {
if( ! pn ) return;
findKthSmallest( pn->left_, k, counter, result );
testAndReturn( k, counter, result );
counter += 1;
testAndReturn( k, counter, result );
findKthSmallest( pn->right_, k, counter, result );
testAndReturn( k, counter, result );
}
// Public API function
void findKthSmallest( Bst const * pt, int const k ) {
int counter = 0;
int result = -1; // -1 := not found
findKthSmallest( pt->root_, k, counter, result );
printf("%d-th element: element = %d\n", k, result );
}
Notes (et différences avec la solution de @ prasadvk):
if( counter == k )
le test est requis à trois endroits: (a) après le sous-arbre de gauche, (b) après la racine et (c) après le sous-arbre de droite. Ceci permet de garantir que le kème élément est détecté pour tous les emplacements , c'est-à-dire quel que soit le sous-arbre dans lequel il se trouve.
if( result == -1 )
test requis pour s'assurer que seul l'élément de résultat est imprimé , sinon tous les éléments à partir du kème plus petit jusqu'à la racine sont imprimés.