1526 lines
45 KiB
Rust
1526 lines
45 KiB
Rust
// Copyright 2013 The Rust Project Developers. See the COPYRIGHT
|
|
// file at the top-level directory of this distribution and at
|
|
// http://rust-lang.org/COPYRIGHT.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
|
|
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
|
|
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
|
|
// option. This file may not be copied, modified, or distributed
|
|
// except according to those terms.
|
|
|
|
//! An ordered map and set implemented as self-balancing binary search
|
|
//! trees. The only requirement for the types is that the key implements
|
|
//! `TotalOrd`.
|
|
|
|
|
|
use std::util::{swap, replace};
|
|
use std::iterator::{FromIterator, Extendable, Peekable};
|
|
use std::cmp::Ordering;
|
|
|
|
// This is implemented as an AA tree, which is a simplified variation of
|
|
// a red-black tree where red (horizontal) nodes can only be added
|
|
// as a right child. The time complexity is the same, and re-balancing
|
|
// operations are more frequent but also cheaper.
|
|
|
|
// Future improvements:
|
|
|
|
// range search - O(log n) retrieval of an iterator from some key
|
|
|
|
// (possibly) implement the overloads Python does for sets:
|
|
// * intersection: &
|
|
// * difference: -
|
|
// * symmetric difference: ^
|
|
// * union: |
|
|
// These would be convenient since the methods work like `each`
|
|
|
|
#[allow(missing_doc)]
|
|
#[deriving(Clone)]
|
|
pub struct TreeMap<K, V> {
|
|
priv root: Option<~TreeNode<K, V>>,
|
|
priv length: uint
|
|
}
|
|
|
|
impl<K: Eq + TotalOrd, V: Eq> Eq for TreeMap<K, V> {
|
|
fn eq(&self, other: &TreeMap<K, V>) -> bool {
|
|
self.len() == other.len() &&
|
|
self.iter().zip(other.iter()).all(|(a, b)| a == b)
|
|
}
|
|
}
|
|
|
|
// Lexicographical comparison
|
|
fn lt<K: Ord + TotalOrd, V: Ord>(a: &TreeMap<K, V>,
|
|
b: &TreeMap<K, V>) -> bool {
|
|
// the Zip iterator is as long as the shortest of a and b.
|
|
for ((key_a, value_a), (key_b, value_b)) in a.iter().zip(b.iter()) {
|
|
if *key_a < *key_b { return true; }
|
|
if *key_a > *key_b { return false; }
|
|
if *value_a < *value_b { return true; }
|
|
if *value_a > *value_b { return false; }
|
|
}
|
|
|
|
a.len() < b.len()
|
|
}
|
|
|
|
impl<K: Ord + TotalOrd, V: Ord> Ord for TreeMap<K, V> {
|
|
#[inline]
|
|
fn lt(&self, other: &TreeMap<K, V>) -> bool { lt(self, other) }
|
|
#[inline]
|
|
fn le(&self, other: &TreeMap<K, V>) -> bool { !lt(other, self) }
|
|
#[inline]
|
|
fn ge(&self, other: &TreeMap<K, V>) -> bool { !lt(self, other) }
|
|
#[inline]
|
|
fn gt(&self, other: &TreeMap<K, V>) -> bool { lt(other, self) }
|
|
}
|
|
|
|
impl<K: TotalOrd, V> Container for TreeMap<K, V> {
|
|
/// Return the number of elements in the map
|
|
fn len(&self) -> uint { self.length }
|
|
|
|
/// Return true if the map contains no elements
|
|
fn is_empty(&self) -> bool { self.root.is_none() }
|
|
}
|
|
|
|
impl<K: TotalOrd, V> Mutable for TreeMap<K, V> {
|
|
/// Clear the map, removing all key-value pairs.
|
|
fn clear(&mut self) {
|
|
self.root = None;
|
|
self.length = 0
|
|
}
|
|
}
|
|
|
|
impl<K: TotalOrd, V> Map<K, V> for TreeMap<K, V> {
|
|
/// Return a reference to the value corresponding to the key
|
|
fn find<'a>(&'a self, key: &K) -> Option<&'a V> {
|
|
let mut current: &'a Option<~TreeNode<K, V>> = &self.root;
|
|
loop {
|
|
match *current {
|
|
Some(ref r) => {
|
|
match key.cmp(&r.key) {
|
|
Less => current = &r.left,
|
|
Greater => current = &r.right,
|
|
Equal => return Some(&r.value)
|
|
}
|
|
}
|
|
None => return None
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
impl<K: TotalOrd, V> MutableMap<K, V> for TreeMap<K, V> {
|
|
/// Return a mutable reference to the value corresponding to the key
|
|
#[inline]
|
|
fn find_mut<'a>(&'a mut self, key: &K) -> Option<&'a mut V> {
|
|
find_mut(&mut self.root, key)
|
|
}
|
|
|
|
/// Insert a key-value pair from the map. If the key already had a value
|
|
/// present in the map, that value is returned. Otherwise None is returned.
|
|
fn swap(&mut self, key: K, value: V) -> Option<V> {
|
|
let ret = insert(&mut self.root, key, value);
|
|
if ret.is_none() { self.length += 1 }
|
|
ret
|
|
}
|
|
|
|
/// Removes a key from the map, returning the value at the key if the key
|
|
/// was previously in the map.
|
|
fn pop(&mut self, key: &K) -> Option<V> {
|
|
let ret = remove(&mut self.root, key);
|
|
if ret.is_some() { self.length -= 1 }
|
|
ret
|
|
}
|
|
}
|
|
|
|
impl<K: TotalOrd, V> TreeMap<K, V> {
|
|
/// Create an empty TreeMap
|
|
pub fn new() -> TreeMap<K, V> { TreeMap{root: None, length: 0} }
|
|
|
|
/// Iterate over the map and mutate the contained values
|
|
pub fn mutate_values(&mut self, f: &fn(&K, &mut V) -> bool) -> bool {
|
|
mutate_values(&mut self.root, f)
|
|
}
|
|
|
|
/// Get a lazy iterator over the key-value pairs in the map.
|
|
/// Requires that it be frozen (immutable).
|
|
pub fn iter<'a>(&'a self) -> TreeMapIterator<'a, K, V> {
|
|
TreeMapIterator {
|
|
stack: ~[],
|
|
node: &self.root,
|
|
remaining_min: self.length,
|
|
remaining_max: self.length
|
|
}
|
|
}
|
|
|
|
/// Get a lazy reverse iterator over the key-value pairs in the map.
|
|
/// Requires that it be frozen (immutable).
|
|
pub fn rev_iter<'a>(&'a self) -> TreeMapRevIterator<'a, K, V> {
|
|
TreeMapRevIterator{iter: self.iter()}
|
|
}
|
|
|
|
/// Get a lazy iterator that should be initialized using
|
|
/// `iter_traverse_left`/`iter_traverse_right`/`iter_traverse_complete`.
|
|
fn iter_for_traversal<'a>(&'a self) -> TreeMapIterator<'a, K, V> {
|
|
TreeMapIterator {
|
|
stack: ~[],
|
|
node: &self.root,
|
|
remaining_min: 0,
|
|
remaining_max: self.length
|
|
}
|
|
}
|
|
|
|
/// Return a lazy iterator to the first key-value pair whose key is not less than `k`
|
|
/// If all keys in map are less than `k` an empty iterator is returned.
|
|
pub fn lower_bound_iter<'a>(&'a self, k: &K) -> TreeMapIterator<'a, K, V> {
|
|
let mut iter: TreeMapIterator<'a, K, V> = self.iter_for_traversal();
|
|
loop {
|
|
match *iter.node {
|
|
Some(ref r) => {
|
|
match k.cmp(&r.key) {
|
|
Less => iter_traverse_left(&mut iter),
|
|
Greater => iter_traverse_right(&mut iter),
|
|
Equal => {
|
|
iter_traverse_complete(&mut iter);
|
|
return iter;
|
|
}
|
|
}
|
|
}
|
|
None => {
|
|
iter_traverse_complete(&mut iter);
|
|
return iter;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/// Return a lazy iterator to the first key-value pair whose key is greater than `k`
|
|
/// If all keys in map are not greater than `k` an empty iterator is returned.
|
|
pub fn upper_bound_iter<'a>(&'a self, k: &K) -> TreeMapIterator<'a, K, V> {
|
|
let mut iter: TreeMapIterator<'a, K, V> = self.iter_for_traversal();
|
|
loop {
|
|
match *iter.node {
|
|
Some(ref r) => {
|
|
match k.cmp(&r.key) {
|
|
Less => iter_traverse_left(&mut iter),
|
|
Greater => iter_traverse_right(&mut iter),
|
|
Equal => iter_traverse_right(&mut iter)
|
|
}
|
|
}
|
|
None => {
|
|
iter_traverse_complete(&mut iter);
|
|
return iter;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/// Get a lazy iterator that consumes the treemap.
|
|
pub fn move_iter(self) -> TreeMapMoveIterator<K, V> {
|
|
let TreeMap { root: root, length: length } = self;
|
|
let stk = match root {
|
|
None => ~[],
|
|
Some(~tn) => ~[tn]
|
|
};
|
|
TreeMapMoveIterator {
|
|
stack: stk,
|
|
remaining: length
|
|
}
|
|
}
|
|
}
|
|
|
|
/// Lazy forward iterator over a map
|
|
pub struct TreeMapIterator<'self, K, V> {
|
|
priv stack: ~[&'self ~TreeNode<K, V>],
|
|
priv node: &'self Option<~TreeNode<K, V>>,
|
|
priv remaining_min: uint,
|
|
priv remaining_max: uint
|
|
}
|
|
|
|
impl<'self, K, V> TreeMapIterator<'self, K, V> {
|
|
#[inline(always)]
|
|
fn next_(&mut self, forward: bool) -> Option<(&'self K, &'self V)> {
|
|
while !self.stack.is_empty() || self.node.is_some() {
|
|
match *self.node {
|
|
Some(ref x) => {
|
|
self.stack.push(x);
|
|
self.node = if forward { &x.left } else { &x.right };
|
|
}
|
|
None => {
|
|
let res = self.stack.pop();
|
|
self.node = if forward { &res.right } else { &res.left };
|
|
self.remaining_max -= 1;
|
|
if self.remaining_min > 0 {
|
|
self.remaining_min -= 1;
|
|
}
|
|
return Some((&res.key, &res.value));
|
|
}
|
|
}
|
|
}
|
|
None
|
|
}
|
|
}
|
|
|
|
impl<'self, K, V> Iterator<(&'self K, &'self V)> for TreeMapIterator<'self, K, V> {
|
|
/// Advance the iterator to the next node (in order) and return a
|
|
/// tuple with a reference to the key and value. If there are no
|
|
/// more nodes, return `None`.
|
|
fn next(&mut self) -> Option<(&'self K, &'self V)> {
|
|
self.next_(true)
|
|
}
|
|
|
|
#[inline]
|
|
fn size_hint(&self) -> (uint, Option<uint>) {
|
|
(self.remaining_min, Some(self.remaining_max))
|
|
}
|
|
}
|
|
|
|
/// Lazy backward iterator over a map
|
|
pub struct TreeMapRevIterator<'self, K, V> {
|
|
priv iter: TreeMapIterator<'self, K, V>,
|
|
}
|
|
|
|
impl<'self, K, V> Iterator<(&'self K, &'self V)> for TreeMapRevIterator<'self, K, V> {
|
|
/// Advance the iterator to the next node (in order) and return a
|
|
/// tuple with a reference to the key and value. If there are no
|
|
/// more nodes, return `None`.
|
|
fn next(&mut self) -> Option<(&'self K, &'self V)> {
|
|
self.iter.next_(false)
|
|
}
|
|
|
|
#[inline]
|
|
fn size_hint(&self) -> (uint, Option<uint>) {
|
|
self.iter.size_hint()
|
|
}
|
|
}
|
|
|
|
/// iter_traverse_left, iter_traverse_right and iter_traverse_complete are used to
|
|
/// initialize TreeMapIterator pointing to element inside tree structure.
|
|
///
|
|
/// They should be used in following manner:
|
|
/// - create iterator using TreeMap::iter_for_traversal
|
|
/// - find required node using `iter_traverse_left`/`iter_traverse_right`
|
|
/// (current node is `TreeMapIterator::node` field)
|
|
/// - complete initialization with `iter_traverse_complete`
|
|
#[inline]
|
|
fn iter_traverse_left<'a, K, V>(it: &mut TreeMapIterator<'a, K, V>) {
|
|
let node = it.node.get_ref();
|
|
it.stack.push(node);
|
|
it.node = &node.left;
|
|
}
|
|
|
|
#[inline]
|
|
fn iter_traverse_right<'a, K, V>(it: &mut TreeMapIterator<'a, K, V>) {
|
|
it.node = &(it.node.get_ref().right);
|
|
}
|
|
|
|
/// iter_traverse_left, iter_traverse_right and iter_traverse_complete are used to
|
|
/// initialize TreeMapIterator pointing to element inside tree structure.
|
|
///
|
|
/// Completes traversal. Should be called before using iterator.
|
|
/// Iteration will start from `self.node`.
|
|
/// If `self.node` is None iteration will start from last node from which we
|
|
/// traversed left.
|
|
#[inline]
|
|
fn iter_traverse_complete<'a, K, V>(it: &mut TreeMapIterator<'a, K, V>) {
|
|
static none: Option<~TreeNode<K, V>> = None;
|
|
match *it.node {
|
|
Some(ref n) => {
|
|
it.stack.push(n);
|
|
it.node = &none;
|
|
}
|
|
None => ()
|
|
}
|
|
}
|
|
|
|
/// Lazy forward iterator over a map that consumes the map while iterating
|
|
pub struct TreeMapMoveIterator<K, V> {
|
|
priv stack: ~[TreeNode<K, V>],
|
|
priv remaining: uint
|
|
}
|
|
|
|
impl<K, V> Iterator<(K, V)> for TreeMapMoveIterator<K,V> {
|
|
#[inline]
|
|
fn next(&mut self) -> Option<(K, V)> {
|
|
while !self.stack.is_empty() {
|
|
let TreeNode {
|
|
key: key,
|
|
value: value,
|
|
left: left,
|
|
right: right,
|
|
level: level
|
|
} = self.stack.pop();
|
|
|
|
match left {
|
|
Some(~left) => {
|
|
let n = TreeNode {
|
|
key: key,
|
|
value: value,
|
|
left: None,
|
|
right: right,
|
|
level: level
|
|
};
|
|
self.stack.push(n);
|
|
self.stack.push(left);
|
|
}
|
|
None => {
|
|
match right {
|
|
Some(~right) => self.stack.push(right),
|
|
None => ()
|
|
}
|
|
self.remaining -= 1;
|
|
return Some((key, value))
|
|
}
|
|
}
|
|
}
|
|
None
|
|
}
|
|
|
|
#[inline]
|
|
fn size_hint(&self) -> (uint, Option<uint>) {
|
|
(self.remaining, Some(self.remaining))
|
|
}
|
|
|
|
}
|
|
|
|
impl<'self, T> Iterator<&'self T> for TreeSetIterator<'self, T> {
|
|
/// Advance the iterator to the next node (in order). If there are no more nodes, return `None`.
|
|
#[inline]
|
|
fn next(&mut self) -> Option<&'self T> {
|
|
do self.iter.next().map_move |(value, _)| { value }
|
|
}
|
|
}
|
|
|
|
impl<'self, T> Iterator<&'self T> for TreeSetRevIterator<'self, T> {
|
|
/// Advance the iterator to the next node (in order). If there are no more nodes, return `None`.
|
|
#[inline]
|
|
fn next(&mut self) -> Option<&'self T> {
|
|
do self.iter.next().map |&(value, _)| { value }
|
|
}
|
|
}
|
|
|
|
/// A implementation of the `Set` trait on top of the `TreeMap` container. The
|
|
/// only requirement is that the type of the elements contained ascribes to the
|
|
/// `TotalOrd` trait.
|
|
pub struct TreeSet<T> {
|
|
priv map: TreeMap<T, ()>
|
|
}
|
|
|
|
impl<T: Eq + TotalOrd> Eq for TreeSet<T> {
|
|
#[inline]
|
|
fn eq(&self, other: &TreeSet<T>) -> bool { self.map == other.map }
|
|
#[inline]
|
|
fn ne(&self, other: &TreeSet<T>) -> bool { self.map != other.map }
|
|
}
|
|
|
|
impl<T: Ord + TotalOrd> Ord for TreeSet<T> {
|
|
#[inline]
|
|
fn lt(&self, other: &TreeSet<T>) -> bool { self.map < other.map }
|
|
#[inline]
|
|
fn le(&self, other: &TreeSet<T>) -> bool { self.map <= other.map }
|
|
#[inline]
|
|
fn ge(&self, other: &TreeSet<T>) -> bool { self.map >= other.map }
|
|
#[inline]
|
|
fn gt(&self, other: &TreeSet<T>) -> bool { self.map > other.map }
|
|
}
|
|
|
|
impl<T: TotalOrd> Container for TreeSet<T> {
|
|
/// Return the number of elements in the set
|
|
#[inline]
|
|
fn len(&self) -> uint { self.map.len() }
|
|
|
|
/// Return true if the set contains no elements
|
|
#[inline]
|
|
fn is_empty(&self) -> bool { self.map.is_empty() }
|
|
}
|
|
|
|
impl<T: TotalOrd> Mutable for TreeSet<T> {
|
|
/// Clear the set, removing all values.
|
|
#[inline]
|
|
fn clear(&mut self) { self.map.clear() }
|
|
}
|
|
|
|
impl<T: TotalOrd> Set<T> for TreeSet<T> {
|
|
/// Return true if the set contains a value
|
|
#[inline]
|
|
fn contains(&self, value: &T) -> bool {
|
|
self.map.contains_key(value)
|
|
}
|
|
|
|
/// Return true if the set has no elements in common with `other`.
|
|
/// This is equivalent to checking for an empty intersection.
|
|
fn is_disjoint(&self, other: &TreeSet<T>) -> bool {
|
|
self.intersection(other).next().is_none()
|
|
}
|
|
|
|
/// Return true if the set is a subset of another
|
|
#[inline]
|
|
fn is_subset(&self, other: &TreeSet<T>) -> bool {
|
|
other.is_superset(self)
|
|
}
|
|
|
|
/// Return true if the set is a superset of another
|
|
fn is_superset(&self, other: &TreeSet<T>) -> bool {
|
|
let mut x = self.iter();
|
|
let mut y = other.iter();
|
|
let mut a = x.next();
|
|
let mut b = y.next();
|
|
while b.is_some() {
|
|
if a.is_none() {
|
|
return false
|
|
}
|
|
|
|
let a1 = a.unwrap();
|
|
let b1 = b.unwrap();
|
|
|
|
match a1.cmp(b1) {
|
|
Less => (),
|
|
Greater => return false,
|
|
Equal => b = y.next(),
|
|
}
|
|
|
|
a = x.next();
|
|
}
|
|
true
|
|
}
|
|
}
|
|
|
|
impl<T: TotalOrd> MutableSet<T> for TreeSet<T> {
|
|
/// Add a value to the set. Return true if the value was not already
|
|
/// present in the set.
|
|
#[inline]
|
|
fn insert(&mut self, value: T) -> bool { self.map.insert(value, ()) }
|
|
|
|
/// Remove a value from the set. Return true if the value was
|
|
/// present in the set.
|
|
#[inline]
|
|
fn remove(&mut self, value: &T) -> bool { self.map.remove(value) }
|
|
}
|
|
|
|
impl<T: TotalOrd> TreeSet<T> {
|
|
/// Create an empty TreeSet
|
|
#[inline]
|
|
pub fn new() -> TreeSet<T> { TreeSet{map: TreeMap::new()} }
|
|
|
|
/// Get a lazy iterator over the values in the set.
|
|
/// Requires that it be frozen (immutable).
|
|
#[inline]
|
|
pub fn iter<'a>(&'a self) -> TreeSetIterator<'a, T> {
|
|
TreeSetIterator{iter: self.map.iter()}
|
|
}
|
|
|
|
/// Get a lazy iterator over the values in the set.
|
|
/// Requires that it be frozen (immutable).
|
|
#[inline]
|
|
pub fn rev_iter<'a>(&'a self) -> TreeSetRevIterator<'a, T> {
|
|
TreeSetRevIterator{iter: self.map.rev_iter()}
|
|
}
|
|
|
|
/// Get a lazy iterator pointing to the first value not less than `v` (greater or equal).
|
|
/// If all elements in the set are less than `v` empty iterator is returned.
|
|
#[inline]
|
|
pub fn lower_bound_iter<'a>(&'a self, v: &T) -> TreeSetIterator<'a, T> {
|
|
TreeSetIterator{iter: self.map.lower_bound_iter(v)}
|
|
}
|
|
|
|
/// Get a lazy iterator pointing to the first value greater than `v`.
|
|
/// If all elements in the set are not greater than `v` empty iterator is returned.
|
|
#[inline]
|
|
pub fn upper_bound_iter<'a>(&'a self, v: &T) -> TreeSetIterator<'a, T> {
|
|
TreeSetIterator{iter: self.map.upper_bound_iter(v)}
|
|
}
|
|
|
|
/// Visit the values (in-order) representing the difference
|
|
pub fn difference<'a>(&'a self, other: &'a TreeSet<T>) -> Difference<'a, T> {
|
|
Difference{a: self.iter().peekable(), b: other.iter().peekable()}
|
|
}
|
|
|
|
/// Visit the values (in-order) representing the symmetric difference
|
|
pub fn symmetric_difference<'a>(&'a self, other: &'a TreeSet<T>)
|
|
-> SymDifference<'a, T> {
|
|
SymDifference{a: self.iter().peekable(), b: other.iter().peekable()}
|
|
}
|
|
|
|
/// Visit the values (in-order) representing the intersection
|
|
pub fn intersection<'a>(&'a self, other: &'a TreeSet<T>)
|
|
-> Intersection<'a, T> {
|
|
Intersection{a: self.iter().peekable(), b: other.iter().peekable()}
|
|
}
|
|
|
|
/// Visit the values (in-order) representing the union
|
|
pub fn union<'a>(&'a self, other: &'a TreeSet<T>) -> Union<'a, T> {
|
|
Union{a: self.iter().peekable(), b: other.iter().peekable()}
|
|
}
|
|
}
|
|
|
|
/// Lazy forward iterator over a set
|
|
pub struct TreeSetIterator<'self, T> {
|
|
priv iter: TreeMapIterator<'self, T, ()>
|
|
}
|
|
|
|
/// Lazy backward iterator over a set
|
|
pub struct TreeSetRevIterator<'self, T> {
|
|
priv iter: TreeMapRevIterator<'self, T, ()>
|
|
}
|
|
|
|
/// Lazy iterator producing elements in the set difference (in-order)
|
|
pub struct Difference<'self, T> {
|
|
priv a: Peekable<&'self T, TreeSetIterator<'self, T>>,
|
|
priv b: Peekable<&'self T, TreeSetIterator<'self, T>>,
|
|
}
|
|
|
|
/// Lazy iterator producing elements in the set symmetric difference (in-order)
|
|
pub struct SymDifference<'self, T> {
|
|
priv a: Peekable<&'self T, TreeSetIterator<'self, T>>,
|
|
priv b: Peekable<&'self T, TreeSetIterator<'self, T>>,
|
|
}
|
|
|
|
/// Lazy iterator producing elements in the set intersection (in-order)
|
|
pub struct Intersection<'self, T> {
|
|
priv a: Peekable<&'self T, TreeSetIterator<'self, T>>,
|
|
priv b: Peekable<&'self T, TreeSetIterator<'self, T>>,
|
|
}
|
|
|
|
/// Lazy iterator producing elements in the set intersection (in-order)
|
|
pub struct Union<'self, T> {
|
|
priv a: Peekable<&'self T, TreeSetIterator<'self, T>>,
|
|
priv b: Peekable<&'self T, TreeSetIterator<'self, T>>,
|
|
}
|
|
|
|
/// Compare `x` and `y`, but return `short` if x is None and `long` if y is None
|
|
fn cmp_opt<T: TotalOrd>(x: Option<&T>, y: Option<&T>,
|
|
short: Ordering, long: Ordering) -> Ordering {
|
|
match (x, y) {
|
|
(None , _ ) => short,
|
|
(_ , None ) => long,
|
|
(Some(x1), Some(y1)) => x1.cmp(y1),
|
|
}
|
|
}
|
|
|
|
impl<'self, T: TotalOrd> Iterator<&'self T> for Difference<'self, T> {
|
|
fn next(&mut self) -> Option<&'self T> {
|
|
loop {
|
|
match cmp_opt(self.a.peek(), self.b.peek(), Less, Less) {
|
|
Less => return self.a.next(),
|
|
Equal => { self.a.next(); self.b.next(); }
|
|
Greater => { self.b.next(); }
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
impl<'self, T: TotalOrd> Iterator<&'self T> for SymDifference<'self, T> {
|
|
fn next(&mut self) -> Option<&'self T> {
|
|
loop {
|
|
match cmp_opt(self.a.peek(), self.b.peek(), Greater, Less) {
|
|
Less => return self.a.next(),
|
|
Equal => { self.a.next(); self.b.next(); }
|
|
Greater => return self.b.next(),
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
impl<'self, T: TotalOrd> Iterator<&'self T> for Intersection<'self, T> {
|
|
fn next(&mut self) -> Option<&'self T> {
|
|
loop {
|
|
let o_cmp = match (self.a.peek(), self.b.peek()) {
|
|
(None , _ ) => None,
|
|
(_ , None ) => None,
|
|
(Some(a1), Some(b1)) => Some(a1.cmp(b1)),
|
|
};
|
|
match o_cmp {
|
|
None => return None,
|
|
Some(Less) => { self.a.next(); }
|
|
Some(Equal) => { self.b.next(); return self.a.next() }
|
|
Some(Greater) => { self.b.next(); }
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
impl<'self, T: TotalOrd> Iterator<&'self T> for Union<'self, T> {
|
|
fn next(&mut self) -> Option<&'self T> {
|
|
loop {
|
|
match cmp_opt(self.a.peek(), self.b.peek(), Greater, Less) {
|
|
Less => return self.a.next(),
|
|
Equal => { self.b.next(); return self.a.next() }
|
|
Greater => return self.b.next(),
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
// Nodes keep track of their level in the tree, starting at 1 in the
|
|
// leaves and with a red child sharing the level of the parent.
|
|
#[deriving(Clone)]
|
|
struct TreeNode<K, V> {
|
|
key: K,
|
|
value: V,
|
|
left: Option<~TreeNode<K, V>>,
|
|
right: Option<~TreeNode<K, V>>,
|
|
level: uint
|
|
}
|
|
|
|
impl<K: TotalOrd, V> TreeNode<K, V> {
|
|
/// Creates a new tree node.
|
|
#[inline]
|
|
pub fn new(key: K, value: V) -> TreeNode<K, V> {
|
|
TreeNode{key: key, value: value, left: None, right: None, level: 1}
|
|
}
|
|
}
|
|
|
|
fn mutate_values<'r, K: TotalOrd, V>(node: &'r mut Option<~TreeNode<K, V>>,
|
|
f: &fn(&'r K, &'r mut V) -> bool)
|
|
-> bool {
|
|
match *node {
|
|
Some(~TreeNode{key: ref key, value: ref mut value, left: ref mut left,
|
|
right: ref mut right, _}) => {
|
|
if !mutate_values(left, |k,v| f(k,v)) { return false }
|
|
if !f(key, value) { return false }
|
|
if !mutate_values(right, |k,v| f(k,v)) { return false }
|
|
}
|
|
None => return false
|
|
}
|
|
true
|
|
}
|
|
|
|
// Remove left horizontal link by rotating right
|
|
fn skew<K: TotalOrd, V>(node: &mut ~TreeNode<K, V>) {
|
|
if node.left.map_default(false, |x| x.level == node.level) {
|
|
let mut save = node.left.take_unwrap();
|
|
swap(&mut node.left, &mut save.right); // save.right now None
|
|
swap(node, &mut save);
|
|
node.right = Some(save);
|
|
}
|
|
}
|
|
|
|
// Remove dual horizontal link by rotating left and increasing level of
|
|
// the parent
|
|
fn split<K: TotalOrd, V>(node: &mut ~TreeNode<K, V>) {
|
|
if node.right.map_default(false,
|
|
|x| x.right.map_default(false, |y| y.level == node.level)) {
|
|
let mut save = node.right.take_unwrap();
|
|
swap(&mut node.right, &mut save.left); // save.left now None
|
|
save.level += 1;
|
|
swap(node, &mut save);
|
|
node.left = Some(save);
|
|
}
|
|
}
|
|
|
|
fn find_mut<'r, K: TotalOrd, V>(node: &'r mut Option<~TreeNode<K, V>>,
|
|
key: &K)
|
|
-> Option<&'r mut V> {
|
|
match *node {
|
|
Some(ref mut x) => {
|
|
match key.cmp(&x.key) {
|
|
Less => find_mut(&mut x.left, key),
|
|
Greater => find_mut(&mut x.right, key),
|
|
Equal => Some(&mut x.value),
|
|
}
|
|
}
|
|
None => None
|
|
}
|
|
}
|
|
|
|
fn insert<K: TotalOrd, V>(node: &mut Option<~TreeNode<K, V>>,
|
|
key: K, value: V) -> Option<V> {
|
|
match *node {
|
|
Some(ref mut save) => {
|
|
match key.cmp(&save.key) {
|
|
Less => {
|
|
let inserted = insert(&mut save.left, key, value);
|
|
skew(save);
|
|
split(save);
|
|
inserted
|
|
}
|
|
Greater => {
|
|
let inserted = insert(&mut save.right, key, value);
|
|
skew(save);
|
|
split(save);
|
|
inserted
|
|
}
|
|
Equal => {
|
|
save.key = key;
|
|
Some(replace(&mut save.value, value))
|
|
}
|
|
}
|
|
}
|
|
None => {
|
|
*node = Some(~TreeNode::new(key, value));
|
|
None
|
|
}
|
|
}
|
|
}
|
|
|
|
fn remove<K: TotalOrd, V>(node: &mut Option<~TreeNode<K, V>>,
|
|
key: &K) -> Option<V> {
|
|
fn heir_swap<K: TotalOrd, V>(node: &mut ~TreeNode<K, V>,
|
|
child: &mut Option<~TreeNode<K, V>>) {
|
|
// *could* be done without recursion, but it won't borrow check
|
|
for x in child.mut_iter() {
|
|
if x.right.is_some() {
|
|
heir_swap(node, &mut x.right);
|
|
} else {
|
|
swap(&mut node.key, &mut x.key);
|
|
swap(&mut node.value, &mut x.value);
|
|
}
|
|
}
|
|
}
|
|
|
|
match *node {
|
|
None => {
|
|
return None; // bottom of tree
|
|
}
|
|
Some(ref mut save) => {
|
|
let (ret, rebalance) = match key.cmp(&save.key) {
|
|
Less => (remove(&mut save.left, key), true),
|
|
Greater => (remove(&mut save.right, key), true),
|
|
Equal => {
|
|
if save.left.is_some() {
|
|
if save.right.is_some() {
|
|
let mut left = save.left.take_unwrap();
|
|
if left.right.is_some() {
|
|
heir_swap(save, &mut left.right);
|
|
} else {
|
|
swap(&mut save.key, &mut left.key);
|
|
swap(&mut save.value, &mut left.value);
|
|
}
|
|
save.left = Some(left);
|
|
(remove(&mut save.left, key), true)
|
|
} else {
|
|
let new = save.left.take_unwrap();
|
|
let ~TreeNode{value, _} = replace(save, new);
|
|
*save = save.left.take_unwrap();
|
|
(Some(value), true)
|
|
}
|
|
} else if save.right.is_some() {
|
|
let new = save.right.take_unwrap();
|
|
let ~TreeNode{value, _} = replace(save, new);
|
|
(Some(value), true)
|
|
} else {
|
|
(None, false)
|
|
}
|
|
}
|
|
};
|
|
|
|
if rebalance {
|
|
let left_level = save.left.map_default(0, |x| x.level);
|
|
let right_level = save.right.map_default(0, |x| x.level);
|
|
|
|
// re-balance, if necessary
|
|
if left_level < save.level - 1 || right_level < save.level - 1 {
|
|
save.level -= 1;
|
|
|
|
if right_level > save.level {
|
|
for x in save.right.mut_iter() { x.level = save.level }
|
|
}
|
|
|
|
skew(save);
|
|
|
|
for right in save.right.mut_iter() {
|
|
skew(right);
|
|
for x in right.right.mut_iter() { skew(x) }
|
|
}
|
|
|
|
split(save);
|
|
for x in save.right.mut_iter() { split(x) }
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
}
|
|
}
|
|
return match node.take() {
|
|
Some(~TreeNode{value, _}) => Some(value), None => fail!()
|
|
};
|
|
}
|
|
|
|
impl<K: TotalOrd, V, T: Iterator<(K, V)>> FromIterator<(K, V), T> for TreeMap<K, V> {
|
|
fn from_iterator(iter: &mut T) -> TreeMap<K, V> {
|
|
let mut map = TreeMap::new();
|
|
map.extend(iter);
|
|
map
|
|
}
|
|
}
|
|
|
|
impl<K: TotalOrd, V, T: Iterator<(K, V)>> Extendable<(K, V), T> for TreeMap<K, V> {
|
|
#[inline]
|
|
fn extend(&mut self, iter: &mut T) {
|
|
for (k, v) in *iter {
|
|
self.insert(k, v);
|
|
}
|
|
}
|
|
}
|
|
|
|
impl<T: TotalOrd, Iter: Iterator<T>> FromIterator<T, Iter> for TreeSet<T> {
|
|
fn from_iterator(iter: &mut Iter) -> TreeSet<T> {
|
|
let mut set = TreeSet::new();
|
|
set.extend(iter);
|
|
set
|
|
}
|
|
}
|
|
|
|
impl<T: TotalOrd, Iter: Iterator<T>> Extendable<T, Iter> for TreeSet<T> {
|
|
#[inline]
|
|
fn extend(&mut self, iter: &mut Iter) {
|
|
for elem in *iter {
|
|
self.insert(elem);
|
|
}
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod test_treemap {
|
|
|
|
use super::*;
|
|
|
|
use std::rand::RngUtil;
|
|
use std::rand;
|
|
|
|
#[test]
|
|
fn find_empty() {
|
|
let m = TreeMap::new::<int, int>(); assert!(m.find(&5) == None);
|
|
}
|
|
|
|
#[test]
|
|
fn find_not_found() {
|
|
let mut m = TreeMap::new();
|
|
assert!(m.insert(1, 2));
|
|
assert!(m.insert(5, 3));
|
|
assert!(m.insert(9, 3));
|
|
assert_eq!(m.find(&2), None);
|
|
}
|
|
|
|
#[test]
|
|
fn test_find_mut() {
|
|
let mut m = TreeMap::new();
|
|
assert!(m.insert(1, 12));
|
|
assert!(m.insert(2, 8));
|
|
assert!(m.insert(5, 14));
|
|
let new = 100;
|
|
match m.find_mut(&5) {
|
|
None => fail!(), Some(x) => *x = new
|
|
}
|
|
assert_eq!(m.find(&5), Some(&new));
|
|
}
|
|
|
|
#[test]
|
|
fn insert_replace() {
|
|
let mut m = TreeMap::new();
|
|
assert!(m.insert(5, 2));
|
|
assert!(m.insert(2, 9));
|
|
assert!(!m.insert(2, 11));
|
|
assert_eq!(m.find(&2).unwrap(), &11);
|
|
}
|
|
|
|
#[test]
|
|
fn test_clear() {
|
|
let mut m = TreeMap::new();
|
|
m.clear();
|
|
assert!(m.insert(5, 11));
|
|
assert!(m.insert(12, -3));
|
|
assert!(m.insert(19, 2));
|
|
m.clear();
|
|
assert!(m.find(&5).is_none());
|
|
assert!(m.find(&12).is_none());
|
|
assert!(m.find(&19).is_none());
|
|
assert!(m.is_empty());
|
|
}
|
|
|
|
#[test]
|
|
fn u8_map() {
|
|
let mut m = TreeMap::new();
|
|
|
|
let k1 = "foo".as_bytes();
|
|
let k2 = "bar".as_bytes();
|
|
let v1 = "baz".as_bytes();
|
|
let v2 = "foobar".as_bytes();
|
|
|
|
m.insert(k1.clone(), v1.clone());
|
|
m.insert(k2.clone(), v2.clone());
|
|
|
|
assert_eq!(m.find(&k2), Some(&v2));
|
|
assert_eq!(m.find(&k1), Some(&v1));
|
|
}
|
|
|
|
fn check_equal<K: Eq + TotalOrd, V: Eq>(ctrl: &[(K, V)],
|
|
map: &TreeMap<K, V>) {
|
|
assert_eq!(ctrl.is_empty(), map.is_empty());
|
|
for x in ctrl.iter() {
|
|
let &(ref k, ref v) = x;
|
|
assert!(map.find(k).unwrap() == v)
|
|
}
|
|
for (map_k, map_v) in map.iter() {
|
|
let mut found = false;
|
|
for x in ctrl.iter() {
|
|
let &(ref ctrl_k, ref ctrl_v) = x;
|
|
if *map_k == *ctrl_k {
|
|
assert!(*map_v == *ctrl_v);
|
|
found = true;
|
|
break;
|
|
}
|
|
}
|
|
assert!(found);
|
|
}
|
|
}
|
|
|
|
fn check_left<K: TotalOrd, V>(node: &Option<~TreeNode<K, V>>,
|
|
parent: &~TreeNode<K, V>) {
|
|
match *node {
|
|
Some(ref r) => {
|
|
assert_eq!(r.key.cmp(&parent.key), Less);
|
|
assert!(r.level == parent.level - 1); // left is black
|
|
check_left(&r.left, r);
|
|
check_right(&r.right, r, false);
|
|
}
|
|
None => assert!(parent.level == 1) // parent is leaf
|
|
}
|
|
}
|
|
|
|
fn check_right<K: TotalOrd, V>(node: &Option<~TreeNode<K, V>>,
|
|
parent: &~TreeNode<K, V>,
|
|
parent_red: bool) {
|
|
match *node {
|
|
Some(ref r) => {
|
|
assert_eq!(r.key.cmp(&parent.key), Greater);
|
|
let red = r.level == parent.level;
|
|
if parent_red { assert!(!red) } // no dual horizontal links
|
|
// Right red or black
|
|
assert!(red || r.level == parent.level - 1);
|
|
check_left(&r.left, r);
|
|
check_right(&r.right, r, red);
|
|
}
|
|
None => assert!(parent.level == 1) // parent is leaf
|
|
}
|
|
}
|
|
|
|
fn check_structure<K: TotalOrd, V>(map: &TreeMap<K, V>) {
|
|
match map.root {
|
|
Some(ref r) => {
|
|
check_left(&r.left, r);
|
|
check_right(&r.right, r, false);
|
|
}
|
|
None => ()
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_rand_int() {
|
|
let mut map = TreeMap::new::<int, int>();
|
|
let mut ctrl = ~[];
|
|
|
|
check_equal(ctrl, &map);
|
|
assert!(map.find(&5).is_none());
|
|
|
|
let mut rng = rand::IsaacRng::new_seeded(&[42]);
|
|
|
|
do 3.times {
|
|
do 90.times {
|
|
let k = rng.gen();
|
|
let v = rng.gen();
|
|
if !ctrl.iter().any(|x| x == &(k, v)) {
|
|
assert!(map.insert(k, v));
|
|
ctrl.push((k, v));
|
|
check_structure(&map);
|
|
check_equal(ctrl, &map);
|
|
}
|
|
}
|
|
|
|
do 30.times {
|
|
let r = rng.gen_uint_range(0, ctrl.len());
|
|
let (key, _) = ctrl.remove(r);
|
|
assert!(map.remove(&key));
|
|
check_structure(&map);
|
|
check_equal(ctrl, &map);
|
|
}
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_len() {
|
|
let mut m = TreeMap::new();
|
|
assert!(m.insert(3, 6));
|
|
assert_eq!(m.len(), 1);
|
|
assert!(m.insert(0, 0));
|
|
assert_eq!(m.len(), 2);
|
|
assert!(m.insert(4, 8));
|
|
assert_eq!(m.len(), 3);
|
|
assert!(m.remove(&3));
|
|
assert_eq!(m.len(), 2);
|
|
assert!(!m.remove(&5));
|
|
assert_eq!(m.len(), 2);
|
|
assert!(m.insert(2, 4));
|
|
assert_eq!(m.len(), 3);
|
|
assert!(m.insert(1, 2));
|
|
assert_eq!(m.len(), 4);
|
|
}
|
|
|
|
#[test]
|
|
fn test_iterator() {
|
|
let mut m = TreeMap::new();
|
|
|
|
assert!(m.insert(3, 6));
|
|
assert!(m.insert(0, 0));
|
|
assert!(m.insert(4, 8));
|
|
assert!(m.insert(2, 4));
|
|
assert!(m.insert(1, 2));
|
|
|
|
let mut n = 0;
|
|
for (k, v) in m.iter() {
|
|
assert_eq!(*k, n);
|
|
assert_eq!(*v, n * 2);
|
|
n += 1;
|
|
}
|
|
assert_eq!(n, 5);
|
|
}
|
|
|
|
#[test]
|
|
fn test_interval_iteration() {
|
|
let mut m = TreeMap::new();
|
|
for i in range(1, 100) {
|
|
assert!(m.insert(i * 2, i * 4));
|
|
}
|
|
|
|
for i in range(1, 198) {
|
|
let mut lb_it = m.lower_bound_iter(&i);
|
|
let (&k, &v) = lb_it.next().unwrap();
|
|
let lb = i + i % 2;
|
|
assert_eq!(lb, k);
|
|
assert_eq!(lb * 2, v);
|
|
|
|
let mut ub_it = m.upper_bound_iter(&i);
|
|
let (&k, &v) = ub_it.next().unwrap();
|
|
let ub = i + 2 - i % 2;
|
|
assert_eq!(ub, k);
|
|
assert_eq!(ub * 2, v);
|
|
}
|
|
let mut end_it = m.lower_bound_iter(&199);
|
|
assert_eq!(end_it.next(), None);
|
|
}
|
|
|
|
#[test]
|
|
fn test_rev_iter() {
|
|
let mut m = TreeMap::new();
|
|
|
|
assert!(m.insert(3, 6));
|
|
assert!(m.insert(0, 0));
|
|
assert!(m.insert(4, 8));
|
|
assert!(m.insert(2, 4));
|
|
assert!(m.insert(1, 2));
|
|
|
|
let mut n = 4;
|
|
for (k, v) in m.rev_iter() {
|
|
assert_eq!(*k, n);
|
|
assert_eq!(*v, n * 2);
|
|
n -= 1;
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_eq() {
|
|
let mut a = TreeMap::new();
|
|
let mut b = TreeMap::new();
|
|
|
|
assert!(a == b);
|
|
assert!(a.insert(0, 5));
|
|
assert!(a != b);
|
|
assert!(b.insert(0, 4));
|
|
assert!(a != b);
|
|
assert!(a.insert(5, 19));
|
|
assert!(a != b);
|
|
assert!(!b.insert(0, 5));
|
|
assert!(a != b);
|
|
assert!(b.insert(5, 19));
|
|
assert!(a == b);
|
|
}
|
|
|
|
#[test]
|
|
fn test_lt() {
|
|
let mut a = TreeMap::new();
|
|
let mut b = TreeMap::new();
|
|
|
|
assert!(!(a < b) && !(b < a));
|
|
assert!(b.insert(0, 5));
|
|
assert!(a < b);
|
|
assert!(a.insert(0, 7));
|
|
assert!(!(a < b) && b < a);
|
|
assert!(b.insert(-2, 0));
|
|
assert!(b < a);
|
|
assert!(a.insert(-5, 2));
|
|
assert!(a < b);
|
|
assert!(a.insert(6, 2));
|
|
assert!(a < b && !(b < a));
|
|
}
|
|
|
|
#[test]
|
|
fn test_ord() {
|
|
let mut a = TreeMap::new();
|
|
let mut b = TreeMap::new();
|
|
|
|
assert!(a <= b && a >= b);
|
|
assert!(a.insert(1, 1));
|
|
assert!(a > b && a >= b);
|
|
assert!(b < a && b <= a);
|
|
assert!(b.insert(2, 2));
|
|
assert!(b > a && b >= a);
|
|
assert!(a < b && a <= b);
|
|
}
|
|
|
|
#[test]
|
|
fn test_lazy_iterator() {
|
|
let mut m = TreeMap::new();
|
|
let (x1, y1) = (2, 5);
|
|
let (x2, y2) = (9, 12);
|
|
let (x3, y3) = (20, -3);
|
|
let (x4, y4) = (29, 5);
|
|
let (x5, y5) = (103, 3);
|
|
|
|
assert!(m.insert(x1, y1));
|
|
assert!(m.insert(x2, y2));
|
|
assert!(m.insert(x3, y3));
|
|
assert!(m.insert(x4, y4));
|
|
assert!(m.insert(x5, y5));
|
|
|
|
let m = m;
|
|
let mut a = m.iter();
|
|
|
|
assert_eq!(a.next().unwrap(), (&x1, &y1));
|
|
assert_eq!(a.next().unwrap(), (&x2, &y2));
|
|
assert_eq!(a.next().unwrap(), (&x3, &y3));
|
|
assert_eq!(a.next().unwrap(), (&x4, &y4));
|
|
assert_eq!(a.next().unwrap(), (&x5, &y5));
|
|
|
|
assert!(a.next().is_none());
|
|
|
|
let mut b = m.iter();
|
|
|
|
let expected = [(&x1, &y1), (&x2, &y2), (&x3, &y3), (&x4, &y4),
|
|
(&x5, &y5)];
|
|
let mut i = 0;
|
|
|
|
for x in b {
|
|
assert_eq!(expected[i], x);
|
|
i += 1;
|
|
|
|
if i == 2 {
|
|
break
|
|
}
|
|
}
|
|
|
|
for x in b {
|
|
assert_eq!(expected[i], x);
|
|
i += 1;
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_from_iter() {
|
|
let xs = ~[(1, 1), (2, 2), (3, 3), (4, 4), (5, 5), (6, 6)];
|
|
|
|
let map: TreeMap<int, int> = xs.iter().map(|&x| x).collect();
|
|
|
|
for &(k, v) in xs.iter() {
|
|
assert_eq!(map.find(&k), Some(&v));
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod bench {
|
|
|
|
use super::*;
|
|
use test::BenchHarness;
|
|
use container::bench::*;
|
|
|
|
// Find seq
|
|
#[bench]
|
|
pub fn insert_rand_100(bh: &mut BenchHarness) {
|
|
let mut m : TreeMap<uint,uint> = TreeMap::new();
|
|
insert_rand_n(100, &mut m, bh);
|
|
}
|
|
|
|
#[bench]
|
|
pub fn insert_rand_10_000(bh: &mut BenchHarness) {
|
|
let mut m : TreeMap<uint,uint> = TreeMap::new();
|
|
insert_rand_n(10_000, &mut m, bh);
|
|
}
|
|
|
|
// Insert seq
|
|
#[bench]
|
|
pub fn insert_seq_100(bh: &mut BenchHarness) {
|
|
let mut m : TreeMap<uint,uint> = TreeMap::new();
|
|
insert_seq_n(100, &mut m, bh);
|
|
}
|
|
|
|
#[bench]
|
|
pub fn insert_seq_10_000(bh: &mut BenchHarness) {
|
|
let mut m : TreeMap<uint,uint> = TreeMap::new();
|
|
insert_seq_n(10_000, &mut m, bh);
|
|
}
|
|
|
|
// Find rand
|
|
#[bench]
|
|
pub fn find_rand_100(bh: &mut BenchHarness) {
|
|
let mut m : TreeMap<uint,uint> = TreeMap::new();
|
|
find_rand_n(100, &mut m, bh);
|
|
}
|
|
|
|
#[bench]
|
|
pub fn find_rand_10_000(bh: &mut BenchHarness) {
|
|
let mut m : TreeMap<uint,uint> = TreeMap::new();
|
|
find_rand_n(10_000, &mut m, bh);
|
|
}
|
|
|
|
// Find seq
|
|
#[bench]
|
|
pub fn find_seq_100(bh: &mut BenchHarness) {
|
|
let mut m : TreeMap<uint,uint> = TreeMap::new();
|
|
find_seq_n(100, &mut m, bh);
|
|
}
|
|
|
|
#[bench]
|
|
pub fn find_seq_10_000(bh: &mut BenchHarness) {
|
|
let mut m : TreeMap<uint,uint> = TreeMap::new();
|
|
find_seq_n(10_000, &mut m, bh);
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod test_set {
|
|
|
|
use super::*;
|
|
|
|
#[test]
|
|
fn test_clear() {
|
|
let mut s = TreeSet::new();
|
|
s.clear();
|
|
assert!(s.insert(5));
|
|
assert!(s.insert(12));
|
|
assert!(s.insert(19));
|
|
s.clear();
|
|
assert!(!s.contains(&5));
|
|
assert!(!s.contains(&12));
|
|
assert!(!s.contains(&19));
|
|
assert!(s.is_empty());
|
|
}
|
|
|
|
#[test]
|
|
fn test_disjoint() {
|
|
let mut xs = TreeSet::new();
|
|
let mut ys = TreeSet::new();
|
|
assert!(xs.is_disjoint(&ys));
|
|
assert!(ys.is_disjoint(&xs));
|
|
assert!(xs.insert(5));
|
|
assert!(ys.insert(11));
|
|
assert!(xs.is_disjoint(&ys));
|
|
assert!(ys.is_disjoint(&xs));
|
|
assert!(xs.insert(7));
|
|
assert!(xs.insert(19));
|
|
assert!(xs.insert(4));
|
|
assert!(ys.insert(2));
|
|
assert!(ys.insert(-11));
|
|
assert!(xs.is_disjoint(&ys));
|
|
assert!(ys.is_disjoint(&xs));
|
|
assert!(ys.insert(7));
|
|
assert!(!xs.is_disjoint(&ys));
|
|
assert!(!ys.is_disjoint(&xs));
|
|
}
|
|
|
|
#[test]
|
|
fn test_subset_and_superset() {
|
|
let mut a = TreeSet::new();
|
|
assert!(a.insert(0));
|
|
assert!(a.insert(5));
|
|
assert!(a.insert(11));
|
|
assert!(a.insert(7));
|
|
|
|
let mut b = TreeSet::new();
|
|
assert!(b.insert(0));
|
|
assert!(b.insert(7));
|
|
assert!(b.insert(19));
|
|
assert!(b.insert(250));
|
|
assert!(b.insert(11));
|
|
assert!(b.insert(200));
|
|
|
|
assert!(!a.is_subset(&b));
|
|
assert!(!a.is_superset(&b));
|
|
assert!(!b.is_subset(&a));
|
|
assert!(!b.is_superset(&a));
|
|
|
|
assert!(b.insert(5));
|
|
|
|
assert!(a.is_subset(&b));
|
|
assert!(!a.is_superset(&b));
|
|
assert!(!b.is_subset(&a));
|
|
assert!(b.is_superset(&a));
|
|
}
|
|
|
|
#[test]
|
|
fn test_iterator() {
|
|
let mut m = TreeSet::new();
|
|
|
|
assert!(m.insert(3));
|
|
assert!(m.insert(0));
|
|
assert!(m.insert(4));
|
|
assert!(m.insert(2));
|
|
assert!(m.insert(1));
|
|
|
|
let mut n = 0;
|
|
for x in m.iter() {
|
|
assert_eq!(*x, n);
|
|
n += 1
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_rev_iter() {
|
|
let mut m = TreeSet::new();
|
|
|
|
assert!(m.insert(3));
|
|
assert!(m.insert(0));
|
|
assert!(m.insert(4));
|
|
assert!(m.insert(2));
|
|
assert!(m.insert(1));
|
|
|
|
let mut n = 4;
|
|
for x in m.rev_iter() {
|
|
assert_eq!(*x, n);
|
|
n -= 1;
|
|
}
|
|
}
|
|
|
|
fn check(a: &[int], b: &[int], expected: &[int],
|
|
f: &fn(&TreeSet<int>, &TreeSet<int>, f: &fn(&int) -> bool) -> bool) {
|
|
let mut set_a = TreeSet::new();
|
|
let mut set_b = TreeSet::new();
|
|
|
|
for x in a.iter() { assert!(set_a.insert(*x)) }
|
|
for y in b.iter() { assert!(set_b.insert(*y)) }
|
|
|
|
let mut i = 0;
|
|
do f(&set_a, &set_b) |x| {
|
|
assert_eq!(*x, expected[i]);
|
|
i += 1;
|
|
true
|
|
};
|
|
assert_eq!(i, expected.len());
|
|
}
|
|
|
|
#[test]
|
|
fn test_intersection() {
|
|
fn check_intersection(a: &[int], b: &[int], expected: &[int]) {
|
|
check(a, b, expected, |x, y, f| x.intersection(y).advance(f))
|
|
}
|
|
|
|
check_intersection([], [], []);
|
|
check_intersection([1, 2, 3], [], []);
|
|
check_intersection([], [1, 2, 3], []);
|
|
check_intersection([2], [1, 2, 3], [2]);
|
|
check_intersection([1, 2, 3], [2], [2]);
|
|
check_intersection([11, 1, 3, 77, 103, 5, -5],
|
|
[2, 11, 77, -9, -42, 5, 3],
|
|
[3, 5, 11, 77]);
|
|
}
|
|
|
|
#[test]
|
|
fn test_difference() {
|
|
fn check_difference(a: &[int], b: &[int], expected: &[int]) {
|
|
check(a, b, expected, |x, y, f| x.difference(y).advance(f))
|
|
}
|
|
|
|
check_difference([], [], []);
|
|
check_difference([1, 12], [], [1, 12]);
|
|
check_difference([], [1, 2, 3, 9], []);
|
|
check_difference([1, 3, 5, 9, 11],
|
|
[3, 9],
|
|
[1, 5, 11]);
|
|
check_difference([-5, 11, 22, 33, 40, 42],
|
|
[-12, -5, 14, 23, 34, 38, 39, 50],
|
|
[11, 22, 33, 40, 42]);
|
|
}
|
|
|
|
#[test]
|
|
fn test_symmetric_difference() {
|
|
fn check_symmetric_difference(a: &[int], b: &[int],
|
|
expected: &[int]) {
|
|
check(a, b, expected, |x, y, f| x.symmetric_difference(y).advance(f))
|
|
}
|
|
|
|
check_symmetric_difference([], [], []);
|
|
check_symmetric_difference([1, 2, 3], [2], [1, 3]);
|
|
check_symmetric_difference([2], [1, 2, 3], [1, 3]);
|
|
check_symmetric_difference([1, 3, 5, 9, 11],
|
|
[-2, 3, 9, 14, 22],
|
|
[-2, 1, 5, 11, 14, 22]);
|
|
}
|
|
|
|
#[test]
|
|
fn test_union() {
|
|
fn check_union(a: &[int], b: &[int],
|
|
expected: &[int]) {
|
|
check(a, b, expected, |x, y, f| x.union(y).advance(f))
|
|
}
|
|
|
|
check_union([], [], []);
|
|
check_union([1, 2, 3], [2], [1, 2, 3]);
|
|
check_union([2], [1, 2, 3], [1, 2, 3]);
|
|
check_union([1, 3, 5, 9, 11, 16, 19, 24],
|
|
[-2, 1, 5, 9, 13, 19],
|
|
[-2, 1, 3, 5, 9, 11, 13, 16, 19, 24]);
|
|
}
|
|
|
|
#[test]
|
|
fn test_zip() {
|
|
let mut x = TreeSet::new();
|
|
x.insert(5u);
|
|
x.insert(12u);
|
|
x.insert(11u);
|
|
|
|
let mut y = TreeSet::new();
|
|
y.insert("foo");
|
|
y.insert("bar");
|
|
|
|
let x = x;
|
|
let y = y;
|
|
let mut z = x.iter().zip(y.iter());
|
|
|
|
// FIXME: #5801: this needs a type hint to compile...
|
|
let result: Option<(&uint, & &'static str)> = z.next();
|
|
assert_eq!(result.unwrap(), (&5u, & &"bar"));
|
|
|
|
let result: Option<(&uint, & &'static str)> = z.next();
|
|
assert_eq!(result.unwrap(), (&11u, & &"foo"));
|
|
|
|
let result: Option<(&uint, & &'static str)> = z.next();
|
|
assert!(result.is_none());
|
|
}
|
|
|
|
#[test]
|
|
fn test_swap() {
|
|
let mut m = TreeMap::new();
|
|
assert_eq!(m.swap(1, 2), None);
|
|
assert_eq!(m.swap(1, 3), Some(2));
|
|
assert_eq!(m.swap(1, 4), Some(3));
|
|
}
|
|
|
|
#[test]
|
|
fn test_pop() {
|
|
let mut m = TreeMap::new();
|
|
m.insert(1, 2);
|
|
assert_eq!(m.pop(&1), Some(2));
|
|
assert_eq!(m.pop(&1), None);
|
|
}
|
|
|
|
#[test]
|
|
fn test_from_iter() {
|
|
let xs = ~[1, 2, 3, 4, 5, 6, 7, 8, 9];
|
|
|
|
let set: TreeSet<int> = xs.iter().map(|&x| x).collect();
|
|
|
|
for x in xs.iter() {
|
|
assert!(set.contains(x));
|
|
}
|
|
}
|
|
}
|