add tree init func: initWithArray

knewbie · Apr 17, 2017 · 172011b · 172011b
1 parent a846764
commit 172011b
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 4 deletions.
diff --git a/Tree.ts b/Tree.ts
@@ -17,6 +17,12 @@ export class BST<T> {
         this.root = null;
     }
 
+    public initWithArray(arr:T[]) {
+        for(let a of arr) {
+            this.insert(a);
+        }
+    }
+
     public insert(d: T) {
         let node = new TreeNode(d);
         if (this.root == null) {

diff --git a/test_Tree.ts b/test_Tree.ts
@@ -2,13 +2,16 @@ import { BST, TreeNode} from "./Tree";
 import {print} from "./util";
 
 
+let arr = [23,45,16,37,3,99,22, 26,40,38,42];
+let arr1 = ['b', 'a','g','2','h','z','y','7','n'];
 
 let t = new BST<number>();
+t.initWithArray(arr);
+
+let t1 = new BST<string>();
+t1.initWithArray(arr1);
+print(t1.inOrder(t1.root));
 
-let arr = [23,45,16,37,3,99,22, 26,40,38,42];
-for(let d of arr) {
-    t.insert(d);
-}
 
 print("Inorder traversal: ");
 print(t.inOrder(t.root));
@@ -28,3 +31,5 @@ t.remove(37);
 print(t.inOrder(t.root));
 
 print(t.count(), t.edges());
+
+