Все родители узла в BST?
При печати Binary Search Tree (BST) используется рекурсивная функция (предварительный заказ). Мне нужно распечатать всех родителей (путь формы корень) текущего узла.
Вспомогательная структура данных может (например, дорожка в моем коде) использовать, но я не хочу продолжать node-> путь хранить путь.
4 / \ / \ 2 6 / \ / \ 1 3 5 7
Предположим, я печатаю узлы в строках, используя предварительный ход:
NODE PATH
4 4
2 4,2
1 4,2,1
3 4,2,3
6 4,6
5 4,6,5
7 4,6,7
Я сделал следующее: Работает нормально!
Дорожка заканчивается значением 0 (ноль) в этом коде. И нет никакого значения узла 0 в BST.
void printpath(int* mypath){
while(*mypath)
printf("%d ", *mypath++);
}
void preorder(struct tree *p, int* path){
int *mypath = calloc(sizeof(path)/sizeof(int) + 1 , sizeof(int*));
int* myp=mypath;
if(p!=NULL){
while( *myp++ = *path++ );
--myp;
*myp=p->data;
*(myp+1)=0;
printf("%d PATH ",p->data);
printpath(mypath);
printf("\n");
preorder(p->left, mypath);
preorder(p->right, mypath);
}
free(mypath);
}
Но я не хочу продолжать массив путей поскольку есть много узлов в BST. Может кто-нибудь предложить мне другую структуру данных / или метод? Предложение будет достаточно, но должно быть эффективным.
Решение
Вот старый трюк, который все еще работает: keep the back pointers in the call stack.
struct stacked_list{
struct stacked_list* prev;
struct tree* tree;
};
void printpath_helper(int data, struct stacked_list* path) {
if (!path->prev)
printf("%d PATH ", data);
else
printpath_helper(data, path->prev);
printf("%d ", path->tree->data);
}
void printpath(struct stacked_list* path) {
printpath_helper(path->tree->data, path);
putchar('\n');
}
void preorder_helper(struct stacked_list* path) {
if (path->tree) {
printpath(path);
struct stacked_list child = {path, path->tree->left};
preorder_helper(&child);
child.tree = path->tree->right;
preorder_helper(&child);
}
}
void preorder(struct tree* tree) {
struct stacked_list root = {NULL, tree};
preorder_helper(&root);
}
Каждая рекурсия preorder_helper создает структуру аргумента и передает ее адрес следующей рекурсии, эффективно создавая связанный список аргументов, который printpath_helper может подойти, чтобы на самом деле напечатать путь. Так как вы хотите напечатать путь сверху вниз, printpath_helper необходимо также перевернуть связанный список, чтобы вы удвоили глубину рекурсии функции; если бы вы могли уйти с печатью снизу вверх, printpath_helper может быть простой цикл (или хвостовая рекурсия).
Другие решения
Вы можете использовать один массив, чтобы сохранить родителей текущего узла и передать его при выполнении рекурсии вместо создания нового массива, просто не забудьте возобновить массив после завершения одной рекурсии. Я думаю, что таким образом вы можете сохранить много воспоминаний. Код ниже:
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
struct node
{
int data;
struct node *left, *right;
};
// A utility function to create a node
struct node* newNode( int data )
{
struct node* temp = (struct node *) malloc( sizeof(struct node) );
temp->data = data;
temp->left = temp->right = NULL;
return temp;
}void print_preorder_path(struct node *root,vector<int>& parents)
{
if(root!=NULL)
{
cout<<root->data<<"\t";
for(size_t i=0;i<parents.size();++i)cout<<parents[i]<<",";
cout<<root->data<<endl;
parents.push_back(root->data);
print_preorder_path(root->left,parents);
print_preorder_path(root->right,parents);
parents.pop_back();
}
}
int main()
{
// Let us construct the tree given in the above diagram
struct node *root = newNode(4);
root->left = newNode(2);
root->left->left = newNode(1);
root->left->right = newNode(3);
root->right = newNode(6);
root->right->left = newNode(5);
root->right->right = newNode(7);
vector<int> parents;
cout<<"NODE\tPATH"<<endl;
print_preorder_path(root,parents);
getchar();
return 0;
}
Код написан на stl для простоты, вы можете изменять его по мере необходимости, надеюсь, это поможет!