Question: Given the root pointer to a binary tree, find the sum of all its elements without using recursion.
Input: Sample Tree (Pointer to node 1 is given).
Output: Sum = 15
We discussed how to find the sum of all elements of a binary tree using recursion in this post.
If recursion seems tough to you , you can always go by the iterative approach and use Level Order Traversal of binary tree. While performing a level order traversal, we just need to create a variable that stores the sum of all elements. Upon removing each element from the queue, just add it to the sum. Here is an implementation of the method described above.
int sumOfBinaryTree(struct binaryTreeNode * root)
{
// Level order traversal
struct binaryTreeNode * temp = NULL;
struct queue * Q = NULL;
// Maintain a sum
int sum = 0;
if(root == NULL)
return 0;
Q = enQueue(Q, root);
while(!isQueueEmpty(Q))
{
temp = Q -> front -> data;
// Add it to the sum
sum += temp -> data;
Q = deQueue(Q);
if(temp -> left)
Q = enQueue(Q, temp -> left);
if(temp -> right)
Q = enQueue(Q, temp -> right);
}
// Delete the queue
free(Q);
// Now return the count
return sum;
}
Time Complexity:- O(n)
Space Complexity:- O(n)
Ideone link for the running code:- http://ideone.com/IfbxoG
