# Data Structures | Tree Traversals | Question 4

What does the following function do for a given binary tree?

 `int` `fun(``struct` `node *root) ` `{ ` `   ``if` `(root == NULL) ` `      ``return` `0; ` `   ``if` `(root->left == NULL && root->right == NULL) ` `      ``return` `0; ` `   ``return` `1 + fun(root->left) + fun(root->right); ` `} `

(A) Counts leaf nodes
(B) Counts internal nodes
(C) Returns height where height is defined as number of edges on the path from root to deepest node
(D) Return diameter where diameter is number of edges on the longest path between any two nodes.

Explanation: The function counts internal nodes.
1) If root is NULL or a leaf node, it returns 0.
2) Otherwise returns, 1 plus count of internal nodes in left subtree, plus count of internal nodes in right subtree.

See the following complete program.

 `#include ` ` `  `struct` `node ` `{ ` `  ``int` `key; ` `  ``struct` `node *left, *right; ` `}; ` ` `  `int` `fun(``struct` `node *root) ` `{ ` `   ``if` `(root == NULL) ` `      ``return` `0; ` `   ``if` `(root->left == NULL && root->right == NULL) ` `      ``return` `0; ` `   ``return` `1 + fun(root->left) + fun(root->right); ` `} ` ` `  `/* Helper function that allocates a new node with the ` `   ``given key and NULL left and right pointers. */` `struct` `node* newNode(``int` `key) ` `{ ` `  ``struct` `node* node = (``struct` `node*) ` `                       ``malloc``(``sizeof``(``struct` `node)); ` `  ``node->key = key; ` `  ``node->left = NULL; ` `  ``node->right = NULL; ` ` `  `  ``return``(node); ` `} ` ` `  `/* Driver program to test above functions*/` `int` `main() ` `{ ` ` `  `  ``/* Constructed binary tree is ` `            ``1 ` `          ``/   \ ` `        ``2      3 ` `      ``/  \    / ` `    ``4     5  8 ` `  ``*/` `  ``struct` `node *root = newNode(1); ` `  ``root->left        = newNode(2); ` `  ``root->right       = newNode(3); ` `  ``root->left->left  = newNode(4); ` `  ``root->left->right = newNode(5); ` `  ``root->right->left = newNode(8); ` ` `  `  ``printf``(``"%d"``, fun(root)); ` ` `  `  ``getchar``(); ` `  ``return` `0; ` `} `

Feeling lost in the world of random DSA topics, wasting time without progress? It's time for a change! Join our DSA course, where we'll guide you on an exciting journey to master DSA efficiently and on schedule.
Ready to dive in? Explore our Free Demo Content and join our DSA course, trusted by over 100,000 geeks!

Previous
Next