230. Késimo elemento más pequeño en un BST
Dada la raíz de un árbol de búsqueda binario y un número entero k , devuelve kthk^{th}kt h valor más pequeño (indexado 1) de todos los valores de los nodos en el árbol.
Ejemplo 1:
Entrada: raíz = [3,1,4,null,2], k = 1
Salida: 1
Ejemplo 2:
Entrada: raíz = [5,3,6,2,4,nulo,nulo,1], k = 3
Salida: 3
Restricciones:
- El número de nodos del árbol es n.
- 1 < = k < = n < = 1 0 4 1 <= k <= n <= 10^41<=k<=norte<=1 04
- 0 < = N odo . valor <= 1 0 4 0 <= Valor.nodo <= 10^40<=Nodo . _ _ _ va l _<=1 04
De: Enlace LeetCode : 230. Kth elemento más pequeño en un BST
Solución:
Ideas:
Así es como funcionará el algoritmo:
- Realice un recorrido en orden del BST.
- A medida que recorra los nodos, lleve un recuento de los nodos visitados.
- Cuando el recuento se vuelve igual a k, el valor de ese nodo es el k-ésimo más pequeño. nodo actual y el nodo anterior.
Código:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
typedef struct {
int count;
int value;
} Result;
void inOrderTraversal(struct TreeNode* root, int k, Result* res) {
if (root == NULL || res->count == k) return;
inOrderTraversal(root->left, k, res);
if (++res->count == k) {
res->value = root->val;
}
inOrderTraversal(root->right, k, res);
}
int kthSmallest(struct TreeNode* root, int k) {
Result res = {
0, 0};
inOrderTraversal(root, k, &res);
return res.value;
}