3.2.28 #

private Node _cache;

public override TValue Get(TKey key)
{
    if (_cache != null && _cache.Key.CompareTo(key) == 0)
    {
        return _cache.Value;
    }

    return Get(root, key).Value;
}

protected override Node Get(Node x, TKey key)
{
    if (key == null)
    {
        throw new ArgumentNullException("calls get() with a null key");
    }

    if (x == null)
    {
        return null;
    }
    var cmp = key.CompareTo(x.Key);
    if (cmp < 0)
    {
        return Get(x.Left, key);
    }
    else if (cmp > 0)
    {
        return Get(x.Right, key);
    }
    else
    {
        _cache = x;
        return x;
    }
}

public override void Put(TKey key, TValue value)
{
    if (key == null)
    {
        throw new ArgumentNullException("calls Put() with a null key");
    }

    if (value == null)
    {
        Delete(key);
        return;
    }

    if (_cache != null && _cache.Key.CompareTo(key) == 0)
    {
        _cache.Value = value;
        return;
    }
    root = Put(root, key, value);
}

protected override Node Put(Node x, TKey key, TValue value)
{
    if (x == null)
    {
        _cache = new Node(key, value, 1);
        return _cache;
    }
    var cmp = key.CompareTo(x.Key);
    if (cmp < 0)
        x.Left = Put(x.Left, key, value);
    else if (cmp > 0)
        x.Right = Put(x.Right, key, value);
    else
        x.Value = value;
    x.Size = 1 + Size(x.Left) + Size(x.Right);
    return x;
}