算法导论:10.4-4
对一个含n个结点的任意有根树,写出一个O(n)时间的过程,输出其所有关键字。
该树以左孩子或兄弟表示法存储。
#ifndef _ROOTED_TREE_H_ #define _ROOTED_TREE_H_ /***************************************************************** 算法导论:10.4-4 对一个含n个结点的任意有根树,写出一个O(n)时间的过程,输出其所有关键字。 该树以左孩子或兄弟表示法存储。 ******************************************************************/ #include <iostream> template <class T> class RootedTree { public: class Node{ public: friend class RootedTree < T > ; Node *getLeftChild()const { return _leftChild; } Node *getRightSibiling()const{ return _rightSibiling; } T value; private: Node() :_parent(nullptr), _leftChild(nullptr), _rightSibiling(nullptr){} Node(const T& v) :_parent(nullptr), _leftChild(nullptr), _rightSibiling(nullptr), value(v){} Node* _parent; // 指向本结点的父结点 Node* _leftChild; // 指向本结点的第一个子结点,如果没有子结点,则为 null Node* _rightSibiling; // 指向本结点的下一个兄弟结点,如果之后没有兄弟结点,则为 null }; RootedTree() :_root(nullptr){ } RootedTree(const T& v){ _root = new Node(v); } ~RootedTree(); // 向指定的根结点中增加子结点,返回第一个子结点的指针。 Node* addChilds(Node*, const T*, size_t); inline Node* getRoot() const { return _root; } void print() const; private: // 二叉树的根结点 Node* _root; void freeNodes(const Node* root); void print(const Node*) const; }; template <class T> RootedTree<T>::~RootedTree(){ // 释放所有结点所占空间 if (_root) freeNodes(_root); } template <class T> void RootedTree<T>::freeNodes(const Node* root){ // 释放左孩子结点 if (root->_leftChild) freeNodes(root->_leftChild); // 释放右兄弟结点 if (root->_rightSibiling) freeNodes(root->_rightSibiling); // 释放本结点 delete root; } template <class T> typename RootedTree<T>::Node* RootedTree<T>::addChilds(Node* root, const T* elements, size_t size){ if (size > 0){ // 创建根结点的第一个子结点 auto firstNode = new Node(elements[0]); firstNode->_parent = root; root->_leftChild = firstNode; Node* node = nullptr; for (size_t i = size - 1; i > 0; --i){ auto sibiling = new Node(elements[i]); sibiling->_parent = root; sibiling->_rightSibiling = node; node = sibiling; } firstNode->_rightSibiling = node; return firstNode; } return nullptr; } template <class T> void RootedTree<T>::print() const{ if (_root) { print(_root); } } template <class T> void RootedTree<T>::print(const Node* root) const{ if (root){ //// 输出它的父结点 //if (root->_parent) // std::cout << " [" << root->_parent->value << "] "; std::cout << root->value << " "; // 打印它之后的兄弟 print(root->_rightSibiling); // 打印第一个子结点 print(root->_leftChild); } } #endif
测试例程:
using namespace std; int main(){ int ia1[] = { 10, 20, 30, 40, 50, 60 }; int ia2[] = {300, 310, 320}; int ia3[] = {3000, 3100, 3200}; int ia4[] = { 200, 210, 230, 240 }; RootedTree<int> tree(0); auto child1 = tree.addChilds(tree.getRoot(), ia1, 6); auto child2 = tree.addChilds(child1->getRightSibiling()->getRightSibiling(), ia2, 3); auto child3 = tree.addChilds(child2, ia3, 3); tree.addChilds(child1->getRightSibiling(), ia4, 4); tree.print(); }
时间: 2024-11-08 20:22:11