前言:
21考研,不论能否进复试记录一下准备路上写下的垃圾代码。本来啃《算法笔记》,但是感觉太多了做不完,改做王道机试指南。
题目描述:
二叉排序树,也称为二叉查找树。可以是一颗空树,也可以是一颗具有如下特性的非空二叉树: 1. 若左子树非空,则左子树上所有节点关键字值均不大于根节点的关键字值; 2. 若右子树非空,则右子树上所有节点关键字值均不小于根节点的关键字值; 3. 左、右子树本身也是一颗二叉排序树。 现在给你N个关键字值各不相同的节点,要求你按顺序插入一个初始为空树的二叉排序树中,每次插入后成功后,求相应的父亲节点的关键字值,如果没有父亲节点,则输出-1。
输入描述
输入包含多组测试数据,每组测试数据两行。
第一行,一个数字N(N<=100),表示待插入的节点数。
第二行,N个互不相同的正整数,表示要顺序插入节点的关键字值,这些值不超过10^8。
输出描述:
输出共N行,每次插入节点后,该节点对应的父亲节点的关键字值。
解答
#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<vector>
#include<cmath>
#include<cstring>
#include<queue>
using namespace std;
struct treenode {
int data;
treenode* left;
treenode* right;
treenode(int num):data(num),left(nullptr),right(nullptr){}
};
treenode* insert_output(treenode* root,int num) {
queue<treenode*> Q;
Q.push(root);
treenode* cur;
treenode* next = new treenode(num);
while (!Q.empty()) { //利用队列找到新结点位置
cur = Q.front();
Q.pop();
if (num > cur->data && cur->right != nullptr)
Q.push(cur->right);
else if (num < cur->data && cur->left != nullptr)
Q.push(cur->left);
}
printf("%d\n", cur->data);
if (num > cur->data)
cur->right = next;
else
cur->left = next;
return root;
}
int main()
{
int n;
while (cin >> n) {
if (n != 0) {
vector<int> vi;
int temp;
for (int i = 0; i < n; i++) {
cin >> temp;
vi.push_back(temp);
}
treenode* root = new treenode(vi[0]);
printf("-1\n"); //根节点直接处理掉
for (int i = 1; i < vi.size(); i++)
root = insert_output(root, vi[i]);
}
else
printf("-1\n");
}
return 0;
}