LeetCode //C - 230. Kth elemento más pequeño en un BST

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:

Insertar descripción de la imagen aquí

Entrada: raíz = [3,1,4,null,2], k = 1
Salida: 1

Ejemplo 2:

Insertar descripción de la imagen aquí

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:

  1. Realice un recorrido en orden del BST.
  2. A medida que recorra los nodos, lleve un recuento de los nodos visitados.
  3. 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;
}

Supongo que te gusta

Origin blog.csdn.net/navicheung/article/details/133010595
Recomendado
Clasificación