I'm a rookie in DS .我改进了二叉树树根到任意一节点的路径de算法时间复杂度变为O(n)

[drmao]

////////////======stack.cpp===========/////////
#include <iostream.h>
#include <stdio.h>
#include <stdlib.h>
#define StaElem BiTree
#define Status int
#define BiTElem char
int ii=0;
typedef struct BiTNode{
BiTElem data;
struct BiTNode *lchild,*rchild;
}BiTNode,*BiTree;
//////////////////////////////////////////////
typedef struct StaNode{
StaElem data;
struct StaNode *next;
}StaNode,*StaNodePtr;
typedef struct Stack{
StaNodePtr base,top;
}Stack;
//---------------------------------------------------//
Status InitStack(Stack &s){
if(s.base=s.top=(StaNodePtr)malloc(sizeof(StaNode)))
{s.base->next=0; return 1;}
else return 0;
}
//----------------------------------------------------//
Status Push(Stack &s,StaElem &e){ StaNodePtr push=(StaNodePtr)malloc(sizeof(StaNode));
s.top->data=e; push->next=s.top; s.top=push;
return 1;
}
//----------------------------------------------------//
Status StackEmpty(Stack &s){return s.top==s.base;}
//----------------------------------------------------//
Status GetTop(Stack &s,StaElem &e){
if(StackEmpty(s)){cout<<"\nSTACK EMPTY";return 0;}
e=s.top->next->data; return 1;
}
//----------------------------------------------------//
Status Pop(Stack &s,StaElem &e){StaNodePtr temp=s.top->next;
if(StackEmpty(s)){cout<<"\nSTACK EMPTY";return 0;}
e=temp->data; free(s.top);s.top=temp; return 1;
}
//----------------------------------------------------//
Status ClearStack(Stack &s){StaNodePtr temp=0;
while(s.top!=s.base){ temp=s.top->next; free(s.top); s.top=temp;}return 1;
}
//-----------------------------------------------------//
Status DestroyStack(Stack &s){ClearStack(s); free(s.base); return 1;
}
//-----------------------------------------------------//
//*******************************************************************//

Status PreOrderCreatBiTree(BiTree &t){
if(!ii){cout<<"\nPreOrderCreatBiTree(BiTree &t)\n";++ii;}
BiTElem input; cin>>input;
if(input=='#')t=0;
else { t=(BiTree)malloc(sizeof(BiTNode));
t->data=input;
PreOrderCreatBiTree(t->lchild); PreOrderCreatBiTree(t->rchild);
} return 1;//先序建立一个二叉树。递归。
}/////////////////////////////////////////////
Status PreOrderTraverse_GetAPointer(BiTree &t,BiTElem data,BiTree &p){//cout<<"\nInOrderTraverse(BiTree &t)\n";
if(t){ if(t->data==data){ p=t; return 1;}
PreOrderTraverse_GetAPointer(t->lchild,data,p);
PreOrderTraverse_GetAPointer(t->rchild,data,p);
} return 0;//找到树中一个等于"data"的节点并用指针p指向它。
}///////////////
Status ARoadFromRootToP_version2(BiTree &Root, BiTree &Ptr){
Stack s,road; int tag[100],top=0; //用后续遍历的方法来找从
BiTree p=Root,temp=0; //根节点到任意的节点Ptr的
InitStack(s); InitStack(road); //路径。
if(!Root)return 0; //tag【1000】为一个标志序列
do{ //如果tag[top]＝＝0表示才从left子树回来
while(p){tag[++top]=0; Push(s,p);p=p->lchild; }//如果＝＝1
if(!StackEmpty(s)){//表示从right子树回来了，可以访问了
if(tag[top]==0){GetTop(s,p);//
if(!StackEmpty(s)){
p=p->rchild; tag[top]=1;}
}else
if(tag[top]==1){ Pop(s,temp);
if(temp==Ptr){///print the road;
Push(s,temp);
while(!StackEmpty(s)){ Pop(s,temp);Push(road,temp);
} cout<<"\n路径是:"<<endl;
while(1){ Pop(road,temp); cout<<temp->data; if(!StackEmpty(road))cout<<"-->";
else goto loop;
}
}//print the road;
top--;
}
}
}while(!StackEmpty(s));
loop: DestroyStack(s); DestroyStack(road);
return 1;
}

///////////////////////////////////////////////
int main(){BiTree tree=0,p=0;char x;
PreOrderCreatBiTree(tree); //p指向某个data等于1的节点。
cout<<"\n你想打印从树根到那个节点的路径？";cin>>x;
PreOrderTraverse_GetAPointer(tree,x,p);
ARoadFromRootToP_version2(tree,p);cout<<endl;//打印从根节点到p的路径。
return 1;
}//////////////////////////////////