Question

Write a function to return an average value of floats stored in a binary tree. In C

Answer 1

The below C code is working perfectly and I have created this binary tree with three nodes.
root node =1 ,root->left=2 and root->right=3.

The below program will find the sum of all nodes and also node count.
Finally, it will calculate the average and print the average result.
Output of My Program:

Total Sum : 6
Total Nodes : 3
Average : 2.00

Code:

#include<stdio.h>
#include<stdlib.h>

struct node
{
int data;
struct node *left,*right;
};

struct node *newnode(int data)
{
   struct node *temp=(struct node*)malloc(sizeof(struct node));
   temp->data=data;
   temp->left=NULL;
   temp->right=NULL;
   return temp;
}

struct node *insert(struct node *root,int data)
{
   if(root==NULL)
   {
       return newnode(data);
   }

   if(data < root->data)
   return insert(root->left,data);

   else if(data > root->data)
   return insert(root->right,data);

}

int total_sum;
int total_nodes;
void average_func(struct node *root)
{
   if(root!=NULL)
   {
       average_func(root->left);

       total_sum=total_sum+root->data;
       total_nodes++;
    
       average_func(root->right);
   }
}

int main()
{

   struct node *root=NULL;

   root=insert(root,1);
   root->left=insert(root,2);
   root->right=insert(root,3);

   average_func(root);

   printf("\nTotal Sum : %d",total_sum);
printf("\nTotal Nodes : %d",total_nodes);

float average=(float)(total_sum/total_nodes);

printf("\nAverage : %.2f",average);

}