二叉树之查找指定节点所在的层数

指定的节点用节点存储的数据来代表,采用递归的方法先判断当前节点是否是目标节点,若不是则依次查找左子树和右子树:

放码子:

#include<iostream> #include<stack> using namespace std; typedef struct tr { int data; struct tr *left,*right; }tre,* tree; stack<tree>st; tree creat(tree root,int val) { tree newroot; if(NULL==root) { newroot=new tre; newroot->data=val; newroot->left=newroot->right=NULL; return newroot; } if(val<=root->data) root->left=creat(root->left,val); else root->right=creat(root->right,val); return root; } void level(tree root,int val,int &ans,int lev) {//lev为中间变量,初始值为1,记录层数 if(NULL==root) ans=-1; else if(root->data==val) ans=lev; else { level(root->left,val,ans,lev+1);//在左子树中查找 if(ans==-1) level(root->right,val,ans,lev+1);//在右子树中查找 } } /*void lbr(tree root) { while(NULL!=root || !st.empty()) { if(NULL!=root) { st.push(root); root=root->left; } else { root=st.top(); st.pop(); printf(“%d “,root->data); root=root->right; } } } void blr(tree root) { while(NULL!=root || !st.empty()) { if(NULL!=root) { printf(“%d “,root->data); st.push(root); root=root->left; } else { root=st.top(); st.pop(); root=root->right; } } }*/ int main() { int n; int res; tree tmp; tmp=NULL; while(scanf(“%d”,&n),n+1) tmp=creat(tmp,n); lbr(tmp); printf(“/n”); blr(tmp); printf(“/n”); while(scanf(“%d”,&n)!=EOF) { res=-1; level(tmp,n,res,1); printf(“%d/n”,res); } return false; }

注释部分用于帮助在纸上画出构建的二分搜索树(知道前序和中序就能构建二叉树的说),然后对比检查结果是否正确。

    原文作者:二叉查找树
    原文地址: https://blog.csdn.net/shifuwawa/article/details/4649557
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞