diff --git a/src/libstd/treemap.rs b/src/libstd/treemap.rs index c3cb45a2bfb3..4b02a13583f8 100644 --- a/src/libstd/treemap.rs +++ b/src/libstd/treemap.rs @@ -600,29 +600,28 @@ fn split(node: &mut ~TreeNode) { fn insert(node: &mut Option<~TreeNode>, key: K, value: V) -> bool { - if node.is_none() { - *node = Some(~TreeNode::new(key, value)); - true - } else { - let mut save = node.swap_unwrap(); + match *node { + Some(ref mut save) => { if key < save.key { let inserted = insert(&mut save.left, key, value); - skew(&mut save); - split(&mut save); - *node = Some(save); // re-balance, if necessary + skew(save); + split(save); inserted } else if save.key < key { let inserted = insert(&mut save.right, key, value); - skew(&mut save); - split(&mut save); - *node = Some(save); // re-balance, if necessary + skew(save); + split(save); inserted } else { save.key = key; save.value = value; - *node = Some(save); false } + } + None => { + *node = Some(~TreeNode::new(key, value)); + true + } } }