adding pcu_aa_count

SCOREC · Jan 12, 2015 · d1f07ea · d1f07ea
1 parent 1b1271f
commit d1f07ea
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/pcu/pcu_aa.c b/pcu/pcu_aa.c
@@ -178,3 +178,10 @@ pcu_aa_node* pcu_aa_find(pcu_aa_node* x, pcu_aa_tree t, pcu_aa_less* less)
     return t;
 }
 
+int pcu_aa_count(pcu_aa_tree t)
+{
+  if (pcu_aa_empty(t))
+    return 0;
+  return pcu_aa_count(t->left) + pcu_aa_count(t->right) + 1;
+}
+
diff --git a/pcu/pcu_aa.h b/pcu/pcu_aa.h
@@ -29,5 +29,6 @@ bool pcu_aa_empty(pcu_aa_tree t);
 pcu_aa_node* pcu_aa_insert(pcu_aa_node* x, pcu_aa_tree* t, pcu_aa_less* less);
 pcu_aa_node* pcu_aa_remove(pcu_aa_node* x, pcu_aa_tree* t, pcu_aa_less* less);
 pcu_aa_node* pcu_aa_find(pcu_aa_node* x, pcu_aa_tree t, pcu_aa_less* less);
+int pcu_aa_count(pcu_aa_tree t);
 
 #endif