Hướng giải của Phần tử nhỏ thứ K trong BST

Nhớ rằng hướng dẫn giải này chỉ nên sử dụng khi bế tắc, và tuyệt đối không nên sao chép mã nguồn kèm theo. Hãy tôn trọng tác giả bài tập và người viết hướng dẫn giải.
Nộp mã nguồn lời giải chính thức trước khi giải bài tập đó có thể khiến bạn bị ban.

Tác giả: kienadmin

Hướng dẫn giải

Phép duyệt inorder của BST luôn cho ta dãy các giá trị được sắp xếp theo thứ tự tăng dần. Do đó, phần tử nhỏ thứ \(K\) chính là phần tử thứ \(K\) trong dãy inorder.

Các bước thực hiện:

  1. Xây dựng BST từ dãy đầu vào.
  2. Thực hiện duyệt inorder để thu được dãy tăng dần.
  3. Với mỗi truy vấn \(K\), in ra phần tử thứ \(K - 1\) trong dãy inorder (do chỉ số bắt đầu từ \(0\)).

Độ phức tạp: \(O(N + Q)\) sau khi xây dựng cây.

Mã nguồn C++

#include <bits/stdc++.h>
using namespace std;

struct Node {
    int val;
    Node *left, *right;
    Node(int v) : val(v), left(nullptr), right(nullptr) {}
};

Node* insert(Node* root, int val) {
    if (!root) return new Node(val);
    if (val < root->val) root->left = insert(root->left, val);
    else root->right = insert(root->right, val);
    return root;
}

void inorder(Node* root, vector<int>& v) {
    if (!root) return;
    inorder(root->left, v);
    v.push_back(root->val);
    inorder(root->right, v);
}

int main() {
    int N, Q;
    cin >> N >> Q;
    Node* root = nullptr;
    for (int i = 0; i < N; ++i) {
        int a; cin >> a;
        root = insert(root, a);
    }
    vector<int> v;
    inorder(root, v);
    while (Q--) {
        int k; cin >> k;
        cout << v[k - 1] << "\n";
    }
    return 0;
}

Nhận xét

Không có ý kiến tại thời điểm này.