D. Tree Construction (BST)

Posted 2021-08-31 Harris-H

D. Tree Construction (BST)

构造BST时，$l<x<r$ ，$l$尽可能大，$r$尽可能小。

则要么添加到$l$的右儿子，要么是$r$的左儿子。

但是两者不可能同时没有，因为根据$BST$的特点，$l$要么是$r$的左儿子，$r$要么是$l$的右儿子。

然后用一个set，两个map维护即可。

#include <iostream>
#include <set>
#include <map>

using namespace std;

int main() {
    ios_base::sync_with_stdio(false); cin.tie(0);

    set<int> numbers;
    map<int, int> left;
    map<int, int> right;

    int n, v;
    cin >> n >> v;
    numbers.insert(v);
    for (int i = 0; i < n - 1; i++) {
        cin >> v;
        auto it = numbers.upper_bound(v);
        int result;
        if (it != numbers.end() && left.count(*it) == 0) {
            left[*it] = v;
            result = *it;
        } else {
            it--;
            right[*it] = v;
            result = *it;
        }
        numbers.insert(v);
        cout << result;
        if (i == n - 2) cout << '\\n';
        else cout << ' ';
    }
}