Метод обхода древовидной структуры - PullRequest
0 голосов
/ 31 октября 2019

У меня есть класс Node и класс Tree (с шаблоном), и я хочу написать метод обхода для моего класса Tree, который распечатывает каждыйузел в моей структуре. Но я получаю только детей от корня, и после этого нет никакого отпечатка.


Так вот мой код. Это немного длинно, но рассматриваемый метод называется traverse (), и он помещается непосредственно перед основной функцией.

#include <iostream>
#include <cstddef>
#define N 5

template <typename V>
class Node {
private:
  V _data;
  unsigned short _size;
  Node<V>* _children;

  template <typename U>
  friend std::ostream& operator<< (std::ostream&, const Node<U>&);

public:
  Node();
  Node(V, unsigned short);
  Node(const Node&); // copy constructor
  Node& operator= (Node&); // assignement by copy constructor
  Node (Node&&); // transfer constructor
  Node& operator= (Node&&); // assignement by transfer constructor
  ~Node();
  V getData() const;
  unsigned short getSize() const;
  Node<V>* getChildren();
  void setChild(unsigned short, Node<V>);
};

template <typename V>
Node<V>::Node()
  : _data(0), _size(0), _children(nullptr) {}

template <typename V>
Node<V>::Node(V data, unsigned short size)
  : _data(data), _size(size), _children(new Node<V>[_size]) {
}

template <typename V>
Node<V>::Node (const Node& other)
  : _size(other._size), _data(other._data) {
  _children = new Node<V>[_size];
  for (unsigned short i = 0; i < _size; i++) 
    _children[i] = other._children[i];
}

template <typename V>
Node<V>& Node<V>::operator= (Node& n){
  if (&n != this) {
    delete[] _children;
    _data = n._data; _children = n._children; _size = n._size;
    n._data = 0; n._size = 0; n._children = nullptr;
  }
  return *this;
}

template <typename V>
Node<V>::Node (Node&& n){
  _data = n._data; _size = n._size; _children = n._children;
  n._data = 0; n._size = 0; n._children = nullptr;
}

template <typename V>
Node<V>& Node<V>::operator= (Node&& n){
  if (&n != this) {
    delete[] _children;
    _data = n._data; _children = n._children; _size = n._size;
    n._data = 0; n._size = 0; n._children = nullptr;
  }
  return *this; 
}

template <typename V>
Node<V>::~Node() { delete[] _children; }

template <typename V> // TO DO : move this to class scope ???
V Node<V>::getData() const {return _data;}

template <typename V>
unsigned short Node<V>::getSize() const {return _size;}

template <typename V>
Node<V>* Node<V>::getChildren() {return _children;}

template <typename V>
void Node<V>::setChild(unsigned short index, Node<V> childNode){
  this->_children[index] = childNode;
}

template <typename V>
std::ostream& operator<< (std::ostream& o, const Node<V>& node){
  if (node._data) o << node._data ;
  return o;
}

template <typename V>
class Tree {
private:
  Node<V>* _info;
public:
  Tree();
  Tree(Node<V>*);
  Tree(V, unsigned short);
  ~Tree() = default;
  Node<V>* info();
  void traverse();
  void traverse_process(Node<V>*, unsigned short);
};

template <typename V>
Tree<V>::Tree() 
  : _info(nullptr) {}

template <typename V>
Tree<V>::Tree(Node<V>* newNode)
  : _info(newNode) {}

template <typename V>
Tree<V>::Tree(V data, unsigned short size) {
  Node<V>* node = new Node<V>(data, size);
  _info = node;
}

template <typename V>
Node<V>* Tree<V>::info() { return _info;}

template <typename V>
void Tree<V>::traverse() {
  traverse_process(this->_info, _info->getSize());
}

template <typename V>
void Tree<V>::traverse_process(Node<V>* node, unsigned short size) {
  if (node){
    for (unsigned short i = 0; i < size; i++) {
      if (node->getChildren()[i].getData()){
        std::cout << node->getChildren()[i] << std::endl;
        traverse_process(node->getChildren()[i], size);
      }
    }
  }
}

int main(int argc, char const *argv[]) {
  Node<char> n1('A', N); // root 
  Node<char> n1_1('B',N); // child of n1
  Node<char> n1_2('C', N); // child of n1
  Node<char> n1_3('D', N); // child of n1
  Node<char> n1_4('E', N); // child of n1
  Node<char> n1_1_1('F', N); // child of n1_1
  Node<char> n1_1_2('G', N); // child of n1_1
  Node<char> n1_1_3('H', N); // child of n1_1
  Node<char> n1_2_1('I', N); // child of n1_2
  Node<char> n1_2_2('J', N); // child of n1_2
  Node<char> n1_1_1_1('K', N); // child of n1_1_1
  n1.setChild(0,n1_1);
  n1.setChild(1,n1_2);
  n1.setChild(2,n1_3);
  n1.setChild(4,n1_4);
  n1_1.setChild(1, n1_1_1);
  n1_1.setChild(2, n1_1_2);
  n1_1.setChild(3, n1_1_3);
  n1_2.setChild(2, n1_2_1);
  n1_2.setChild(4, n1_2_2);
  n1_1_1.setChild(0, n1_1_1_1);
  Tree<char> t(&n1);
  t.traverse();
  return 0;
}

Спасибо вам за помощь

1 Ответ

0 голосов
/ 31 октября 2019

У ваших узлов есть две копии:

  • Вы размещаете их здесь в стеке: Node<char> n1('A', N);
  • Они создают своих собственных детей здесь: _children = new Node<V>[_size];

Когда вы вызываете setChild (), он делает это:

  • this->_children[index] = childNode

Который копирует весьузел от одного в вашем стеке до того, который принадлежит узлу

Так что, когда вы добавляете внука к тому в стеке, traverse_process ничего не увидит, потому что использует другую копию,Вот почему первый уровень обхода работает, а более глубокие уровни - нет.

Чтобы исправить это, вам нужно либо добавить средства доступа для дочерних узлов Node, и вызвать setChild туда . ИЛИ вам нужно изменить способ работы _children, чтобы указатели указывали на узлы, выделенные для стека.

...