fixed CSplayTree.size

marsupinole · Apr 1, 2009 · 90c214a · 90c214a
1 parent 5dd3ee5
commit 90c214a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/ext/containers/splaytree_map/splaytree.c b/ext/containers/splaytree_map/splaytree.c
@@ -55,6 +55,7 @@ static splaytree_node* splay(splaytree *tree, splaytree_node *n, VALUE key) {
 				y = n->left;
 				n->left = y->right;
 				y->right = n;
+				n->size = node_size(n->left) + node_size(n->right) + 1;
 				n = y;
           		if (!n->left) break;
 			}
@@ -69,6 +70,7 @@ static splaytree_node* splay(splaytree *tree, splaytree_node *n, VALUE key) {
 				y = n->right;
 				n->right = y->left;
 				y->left = n;
+				n->size = node_size(n->left) + node_size(n->right) + 1;
 				n = y;
           		if (!n->right) break;
 			}