测试

Author: masx200

finding-mk-average/MultiRedBlackTree.ts

@@ -36,16 +36,15 @@ export class MultiRedBlackTree<T> extends RedBlackTreeExtended<T> {
     insertGetNode(value: T): MultiRedBlackNode<T> | null {
         let node = this.insertNode(
             MultiRedBlackNode,
-            value,
+            value
         ) as MultiRedBlackNode<T> | null;
         if (node) {
             while (node.parent?.red) {
                 let parent: MultiRedBlackNode<T> = node.parent!;
-                const parentDirection: Direction = parent
-                    .directionFromParent()!;
-                const uncleDirection: Direction = parentDirection === "right"
-                    ? "left"
-                    : "right";
+                const parentDirection: Direction =
+                    parent.directionFromParent()!;
+                const uncleDirection: Direction =
+                    parentDirection === "right" ? "left" : "right";
                 const uncle: MultiRedBlackNode<T> | null =
                     parent.parent![uncleDirection] ?? null;
 

finding-mk-average/index.ts

@@ -20,9 +20,18 @@ class MKAverage {
         if (this.queue.length === this.m) {
             const willBeDelete: number = this.queue[0];
             this.queue.shift();
-
+            console.log("deleteElement", willBeDelete);
             this.total -= willBeDelete;
             this.tree.remove(willBeDelete);
+            const root = this.tree.getRoot();
+            if (!root) throw Error("null root");
+            {
+                const temp: [number, number][] = [];
+                for (const node of InOrderIterator(root)) {
+                    temp.push([node.value, node.count]);
+                }
+                console.log(JSON.stringify(temp));
+            }
         }
 
         this.queue.push(num);
@@ -31,6 +40,7 @@ class MKAverage {
         this.total += num;
         this.count++;
         // debugger;
+        console.log("addElement", num);
         const root = this.tree.getRoot();
         if (!root) throw Error("null root");
         {
@@ -48,14 +58,14 @@ class MKAverage {
         let ret = this.total;
         const root = this.tree.getRoot();
         if (!root) throw Error("null root");
+        console.log("calculateMKAverage");
         {
             const temp: [number, number][] = [];
             for (const node of InOrderIterator(root)) {
                 temp.push([node.value, node.count]);
             }
             console.log(JSON.stringify(temp));
         }
-       
 
         // console.log(root)
         let k = this.k;