【2020复旦机试】D.搜索二叉树

【输出格式】
⼀⾏ n 个整数，其中第 i 个整数 ai 表⽰元素 i 对应节点的⽗亲节点的元素。特别地，根节点的⽗亲节
点元素视为 0。
【样例输⼊】
5
2 3 5 1 4
【样例输出】
2 0 2 5 3
【样例解释】
最后建出来的⼆叉搜索树如下：
2
/ \
1 3
\
5
/
4
1 的⽗亲为 2，2 为根结点，所以⽗亲为 0，3 的⽗亲为 2，4 的⽗亲为 5，5 的⽗亲为 3。
【时空限制】
5000ms，256MB

解题思路：

参考力扣题目

 1008. 前序遍历构造二叉搜索树

力扣https://leetcode-cn.com/problems/construct-binary-search-tree-from-preorder-traversal/submissions/

先用这种方法构造搜索二叉树

接着递归查找，1~n节点的父节点，注意看出口条件（在代码中），结合搜索树性质，如果不是出口，根据值比较判断是在左子树还是右子树上接着找。

注意打印，我都是默认最后一个数后面没有空格的，PAT上一般也都是这样，所以find函数传了个i和n，i  !=  n时才打印空格，最后一个就是i==n的时候，结尾不要空格。

        if(i!=n){
        cout<<" ";
        }

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

//定义二叉树结构，直接从LeetCode抄 
struct TreeNode {
	int val;
	TreeNode *left;
	TreeNode *right;
	TreeNode() : val(0), left(nullptr), right(nullptr) {}
	TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
	TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
}; 
//插入函数，按照前序插入 
void insert(TreeNode *root,int val){
    if(root->val > val){ //如果当前结点值比插入值大，则应该往左边插入
        if(!root->left)root->left = new TreeNode(val); //如果左边为空，可以直接插入
        else insert(root->left,val); //如果左边不为空，那么继续往左边深入查找可以插入的地方
	}else {
        if(!root->right)root->right = new TreeNode(val);
        else insert(root->right,val);
    }
}
//建树函数 
TreeNode* bstFromPreorder(vector<int> &preorder){
	TreeNode* root = new TreeNode(preorder[0]);
	for(int i = 1; i < preorder.size(); i++ ){
		insert(root, preorder[i]);	
	}
	return root;
}

//查找父节点的函数，打印结果 
void find(TreeNode* root, int val, int& i, int& n ){
	//出口，为空，或者是要查找节点的父节点 
	if(!root) return;
	if(root->left && root->left->val == val || root->right && root->right->val == val){
		cout<<root->val; 
		if(i!=n){
			cout<<" ";
		}
		return;
	}
	//如果不是出口，判断去左子树还是右子树上去查找 
	else if(root->val > val) find(root->left, val, i, n);
	else if(root->val < val) find(root->right,val, i, n);
	//特判根节点，因为如果不是根节点，代码不会走到这里，在上面第二种出口那里已经return了 
	else {
		cout<<"0";
		if(i!=n){
		cout<<" ";
		}
		return;
	}
	

}
//测试用的中序遍历，看看插入的对不对：） 
void inorder(TreeNode* root){
	if(root==nullptr) return;
	inorder(root->left);
	cout<<root->val<<" ";
	inorder(root->right);
	
}


int main(){
	int n;
	cin>>n;
	int i=0;
	int temp;
	vector<int> preorder;
	while(i<n){
		cin >>  temp;
		preorder.push_back(temp);
		i++;
	}
	TreeNode* root = bstFromPreorder(preorder); 
	i=1;
    while(i<=n){
    	find(root,i,i,n);
    	i++;
	}
	

}