Fix nextTreeData

2025-12-07 21:26:44 +01:00 · 2025-12-07 21:26:44 +01:00 · 492a101160
commit 492a101160
parent c79a61e8ee
1 changed files with 4 additions and 2 deletions
--- a/bintree.c
+++ b/bintree.c
@ -77,12 +77,14 @@ void *nextTreeData(TreeNode *root)
        current = root;
    }

-    while (current != NULL)
-    {
+    while (current != NULL){
        stack = push(stack, current);
        current = current->left;
    }

+    if(stack == NULL){
+        return 0;
+    }
    TreeNode *node = (TreeNode *)top(stack);
    stack = pop(stack);