c++友元类，友元函数
#ifndef _BST_H_HITHUMING_INCLUDE_
#define _BST_H_HITHUMING_INCLUDE_
#include<iostream>
using namespace std;
template<class T>
class node
{
public:
node():lchild(NULL),rchild(NULL) {};
node(T p)
{
this->data=p;
lchild=NULL;
rchild=NULL;
};
~node()
{
delete this;
};
void setup(T d,node<T>* l,node<T> r)
{
data=d;
lchild=l;
rchild=r;
};
T return_data()
{
return data;
};
node<T>* return_lc()
{
return lchild?lchild:NULL;
};
node<T>* return_rc()
{
return rchild?rchild:NULL;
};
friend class BST;//声明啦友元类，为什么下面还不能访问？
private:
T data;
node<T>* lchild,rchild;
};
template <class T>
class BST
{
public:
void CreateBST();//返回父节点的指针
void InsertBST(T key,node<T>* F);//插入的类型，还有插入的位置
node<T>* SearchBST(T key,node<T>* F);//查找
void in_order(node<T>* t);
node<T>* return_rt();
private:
node<T>* root;
};
template <class T>
void BST<T>::CreateBST()
{
root=NULL;
T m;
while(cin >> m&&m!='#')
InsertBST(m,root);
}
template <class T>
void BST<T>::InsertBST(T key,node<T>* F)
{
if(F==NULL)
{
// F=new node<T>;
F.setup(key,NULL,NULL);
}
else if (key<F.return_data())
InsertBST(key,F.return_lc());
else if (key > F.return_data())
InsertBST(key,F.return_rc());
else return;
}
template <class T>
node<T>* BST<T>::SearchBST(T k,node<T>* F)
{
node<T>* p =F;
if (p== NULL || k == p.return_data() ) /* 递归终止条件*/
return p;
if ( k < p.return_data())
return (SearchBST (k, p.return_lc())); /* 查找左子树*/
else
return (SearchBST(k,p.return_rc())) ; /* 查找右子树*/
}
template <class T>
void BST<T>::in_order(node<T>* t)
{
if(t!=NULL)
{
in_order(t.return_lc());
cout << t.return_data();
in_order(t.return_rc());
}
}
#endif
#include <iostream>
#include <cstring>
using namespace std;
// 添加BST模板的前置声明，否则node中无法声明BST<T>为友元类
template <class T>
class BST;
template<class T>
class node
{
public:
node():lchild(NULL),rchild(NULL) {};
node(T p)
{
this->data=p;
lchild=NULL;
rchild=NULL;
};
~node()
{
delete this;
};
// 代码这里错了，第三个参数应该是指针类型的
// void setup(T d,node<T>* l,node<T> r)
void setup(T d,node<T>* l,node<T>* r)
{
data=d;
lchild=l;
rchild=r;
};
T return_data()
{
return data;
};
// 这里不需要判断，因为lchild和rchild都有合理默认值的
node<T>* return_lc() { return lchild; };
node<T>* return_rc() { return rchild; };
// friend class BST;//声明啦友元类，为什么下面还不能访问？
// 那是因为仅仅写BST还不够
friend class BST<T>;
private:
T data;
// 这里又错了，这样写rchild不是指针，而是node<T>类型
// node<T>* lchild,rchild;
node<T> *lchild, *rchild;
};
template <class T>
class BST
{
public:
void CreateBST();//返回父节点的指针
void InsertBST(T key,node<T>* F);//插入的类型，还有插入的位置
node<T>* SearchBST(T key,node<T>* F);//查找
void in_order(node<T>* t);
node<T>* return_rt() { return root; }
private:
node<T>* root;
};
template <class T>
void BST<T>::CreateBST()
{
root=NULL;
T m;
while(cin >> m&&m!='#')
InsertBST(m,root);
}
template <class T>
void BST<T>::InsertBST(T key,node<T>* F)
{
// 这里F是指针，怎么能.呢？
if(F==NULL)
{
// F为空怎么还调用？我反正只改语法，能编译就好，细节你再慢慢修改哦
F->setup(key,NULL,NULL);
}
else if (key < F->return_data())
InsertBST(key,F->return_lc());
else if (key > F->return_data())
InsertBST(key,F->return_rc());
else return;
}
template <class T>
node<T>* BST<T>::SearchBST(T k,node<T>* F)
{
node<T>* p =F;
// 这里p也是指针，同样要用->操作符
if (p== NULL
[解决办法]
k == p->return_data() ) /* 递归终止条件*/
return p;
if ( k < p->return_data())
return (SearchBST (k, p->return_lc())); /* 查找左子树*/
else
return (SearchBST(k,p->return_rc())) ; /* 查找右子树*/
}
template <class T>
void BST<T>::in_order(node<T>* t)
{
if(t!=NULL)
{
// 同上
in_order(t->return_lc());
cout << t->return_data();
in_order(t->return_rc());
}
}
int main(int argc, char **argv)
{
BST<int> b;
b.CreateBST();
b.InsertBST(3, new node<int>());
b.SearchBST(3, new node<int>());
b.in_order(new node<int>());
b.return_rt();
return 0;
}
