Возврат адреса указателя из функции в C

Question

Исправлено: Проблема была с рекурсией, а не указателями. Когда он вернется через каждую рекурсию, он вернет возвращаемый VAR к старой maintree. Исправлено, создав глобальный VAR и установив его перед возвратом, а затем выполнил полную функцию return void. Нет утечек памяти.

Я пытаюсь передать адрес указателя malloc'd в переменную-указатель в main().

Код работает до определенной точки. Вместо того, чтобы передавать адрес указателя из FinalTree на главную, он создает новую копию, что означает, что я не могу освободить() память, которая была malloc'd в FinalTree(). Когда я делаю проверку памяти, я остаюсь с некоторой свободной памятью.

Вот код до сих пор:

Tree* FinalTree(Forest** forest, int FinalFreq) 
{ 

    Tree* maintree = mktree(); //this function calls malloc and have to use it 
    Tree* tree1 = pick(*forest); 
    Tree* tree2 = pick(*forest); 

    maintree->frequency = tree1->frequency + tree2->frequency; 
    maintree->left = tree1; 
    maintree->right = tree2; 

    //Make sure forest isnt barren 
    if (maintree->frequency != FinalFreq) 
    { 
     plant(*forest, maintree); 
     FinalTree(forest, FinalFreq); 
    } 

    return maintree; 
} 

Это вызывается из основной() путем:

Tree* tfinal = FinalTree(&finit, header.checksum); 

Указатели вождения меня с ума .. Что мне не хватает?

Answer 1

Пожалуйста, простите меня, если я полностью не понимаю ваш вопрос, но основываясь на комментарии, я думаю, что у меня есть идея, о чем вы сейчас говорите.

Эта функция создает дерево, но возвращает только адрес корневого узла верхнего уровня. Ваша проблема в том, что если вы free(), что один, остальная часть дерева по-прежнему выделена, правильно? Если это так, я думаю, что решение этого заключается в том, чтобы написать выделенную функцию FreeTree(Tree *), которая пересекает дерево, и отдельные узлы листьев на пути назад.

Нечто подобное (не проверено):

void FreeTree(Tree *tree) { 
    if (tree == NULL) 
    return; 

    FreeTree(tree->left); 
    FreeTree(tree->right); 
    free(tree);                            
} 

Update: Ах хорошо. Это действительно странно. Я не вижу ничего в функции, которая изменила бы maintree в стороне от mktree(), который ее создает. Если plant не принимает ссылку на указатель:

void plant(Forest forest, Tree *&tree) 

Затем plant() может изменить указатель maintree указывает. В противном случае похоже, что он должен возвращать корневой узел.