Create 14. Count nodes.cpp

Author: maityamit

008 - Trees/14. Count nodes.cpp

@@ -0,0 +1,35 @@
+/************************************************************
+ 
+ Following is the structure for the TreeNode class
+
+ template <typename T>
+ class TreeNode {
+ public:
+ T data;
+ vector<TreeNode<T>*> children;
+ 
+ TreeNode(T data) {
+ this->data = data;
+ }
+ 
+ ~TreeNode() {
+ for (int i = 0; i < children.size(); i++) {
+ delete children[i];
+ }
+ }
+ };
+
+************************************************************/
+
+int getLargeNodeCount(TreeNode<int>* root, int x) {
+ // Write your code here
+ int count = 0;
+ for(int i=0;i<root->children.size();i++){
+ int temp = getLargeNodeCount(root->children[i],x);
+ count = count+temp;
+ }
+ if(root->data > x){
+ count++;
+ }
+ return count;
+}