#include using namespace std; struct Node { int data; Node* left; Node* right; Node(int val) { data = val; left = nullptr; right = nullptr; } }; Node* construct(vector& arr) { Node* root = new Node(arr[0]); pair p = {root, 1}; stack> st; st.push(p); int idx = 1; while (!st.empty()) { if (st.top().second == 1) { st.top().second++; if (arr[idx] != -1) { st.top().first->left = new Node(arr[idx]); pair lp = {st.top().first->left, 1}; st.push(lp); } else { st.top().first->left = nullptr; } idx++; } else if (st.top().second == 2) { st.top().second++; if (arr[idx] != -1) { st.top().first->right = new Node(arr[idx]); pair rp = {st.top().first->right, 1}; st.push(rp); } else { st.top().first->right = nullptr; } idx++; } else { st.pop(); } } return root; } void display(Node* node) { if (node == nullptr) { return; } string str = " <- " + to_string(node->data) + " -> "; string left = (node->left == nullptr) ? "." : "" + to_string(node->left->data); string right = (node->right == nullptr) ? "." : "" + to_string(node->right->data); str = left + str + right; cout << str << endl; display(node->left); display(node->right); } Node* add(Node* node, int val) { // Write your code here } int main() { int n; cin >> n; vector a(n, 0); for (int i = 0; i < n; i++) { string x; cin >> x; if (x == "n") { a[i] = -1; } else { a[i] = stoi(x); } } int data; cin >> data; Node* root = construct(a); root = add(root, data); display(root); }

Add Node To Bst

Related Questions