NetBSD/gnu/lib/libg++/include/gen/SplayMap.ccP

402 lines
7.8 KiB
COBOL

// This may look like C code, but it is really -*- C++ -*-
/*
Copyright (C) 1988 Free Software Foundation
written by Doug Lea (dl@rocky.oswego.edu)
This file is part of the GNU C++ Library. This library is free
software; you can redistribute it and/or modify it under the terms of
the GNU Library General Public License as published by the Free
Software Foundation; either version 2 of the License, or (at your
option) any later version. This library is distributed in the hope
that it will be useful, but WITHOUT ANY WARRANTY; without even the
implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
PURPOSE. See the GNU Library General Public License for more details.
You should have received a copy of the GNU Library General Public
License along with this library; if not, write to the Free Software
Foundation, 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
*/
#ifdef __GNUG__
#pragma implementation
#endif
#include <stream.h>
#include "<T>.<C>.SplayMap.h"
/*
struct to simulate the special `null' node in the Sleater & Tarjan JACM 1985
splay tree algorithms
All routines use a version of their `simple top-down' splay alg. (p 669)
*/
struct _dummySplayNode
{
<T><C>SplayNode* lt;
<T><C>SplayNode* rt;
<T><C>SplayNode* par;
} _dummy_null;
/*
traversal primitives
*/
<T><C>SplayNode* <T><C>SplayMap::leftmost()
{
<T><C>SplayNode* t = root;
if (t != 0) while (t->lt != 0) t = t->lt;
return t;
}
<T><C>SplayNode* <T><C>SplayMap::rightmost()
{
<T><C>SplayNode* t = root;
if (t != 0) while (t->rt != 0) t = t->rt;
return t;
}
<T><C>SplayNode* <T><C>SplayMap::succ(<T><C>SplayNode* t)
{
if (t == 0)
return 0;
if (t->rt != 0)
{
t = t->rt;
while (t->lt != 0) t = t->lt;
return t;
}
else
{
for (;;)
{
if (t->par == 0 || t == t->par->lt)
return t->par;
else
t = t->par;
}
}
}
<T><C>SplayNode* <T><C>SplayMap::pred(<T><C>SplayNode* t)
{
if (t == 0)
return 0;
else if (t->lt != 0)
{
t = t->lt;
while (t->rt != 0) t = t->rt;
return t;
}
else
{
for (;;)
{
if (t->par == 0 || t == t->par->rt)
return t->par;
else
t = t->par;
}
}
}
Pix <T><C>SplayMap::seek(<T&> key)
{
<T><C>SplayNode* t = root;
if (t == 0)
return 0;
int comp = <T>CMP(key, t->item);
if (comp == 0)
return Pix(t);
<T><C>SplayNode* dummy = (<T><C>SplayNode*)(&_dummy_null);
<T><C>SplayNode* l = dummy;
<T><C>SplayNode* r = dummy;
dummy->rt = dummy->lt = dummy->par = 0;
while (comp != 0)
{
if (comp > 0)
{
<T><C>SplayNode* tr = t->rt;
if (tr == 0)
break;
else
{
comp = <T>CMP(key, tr->item);
if (comp <= 0 || tr->rt == 0)
{
l->rt = t; t->par = l;
l = t;
t = tr;
if (comp >= 0)
break;
}
else
{
if ((t->rt = tr->lt) != 0) t->rt->par = t;
tr->lt = t; t->par = tr;
l->rt = tr; tr->par = l;
l = tr;
t = tr->rt;
comp = <T>CMP(key, t->item);
}
}
}
else
{
<T><C>SplayNode* tl = t->lt;
if (tl == 0)
break;
else
{
comp = <T>CMP(key, tl->item);
if (comp >= 0 || tl->lt == 0)
{
r->lt = t; t->par = r;
r = t;
t = tl;
if (comp <= 0)
break;
}
else
{
if ((t->lt = tl->rt) != 0) t->lt->par = t;
tl->rt = t; t->par = tl;
r->lt = tl; tl->par = r;
r = tl;
t = tl->lt;
comp = <T>CMP(key, t->item);
}
}
}
}
if ((r->lt = t->rt) != 0) r->lt->par = r;
if ((l->rt = t->lt) != 0) l->rt->par = l;
if ((t->lt = dummy->rt) != 0) t->lt->par = t;
if ((t->rt = dummy->lt) != 0) t->rt->par = t;
t->par = 0;
root = t;
return (comp == 0) ? Pix(t) : 0;
}
<C>& <T><C>SplayMap::operator [] (<T&> item)
{
<T><C>SplayNode* t = root;
if (t == 0)
{
++count;
root = new <T><C>SplayNode(item, def);
return root->cont;
}
int comp = <T>CMP(item, t->item);
if (comp == 0)
return t->cont;
<T><C>SplayNode* dummy = (<T><C>SplayNode*)(&_dummy_null);
<T><C>SplayNode* l = dummy;
<T><C>SplayNode* r = dummy;
dummy->rt = dummy->lt = dummy->par = 0;
while (comp != 0)
{
if (comp > 0)
{
<T><C>SplayNode* tr = t->rt;
if (tr == 0)
{
++count;
tr = new <T><C>SplayNode(item, def);
comp = 0;
}
else
comp = <T>CMP(item, tr->item);
if (comp <= 0)
{
l->rt = t; t->par = l;
l = t;
t = tr;
}
else
{
<T><C>SplayNode* trr = tr->rt;
if (trr == 0)
{
++count;
trr = new <T><C>SplayNode(item, def);
comp = 0;
}
else
comp = <T>CMP(item, trr->item);
if ((t->rt = tr->lt) != 0) t->rt->par = t;
tr->lt = t; t->par = tr;
l->rt = tr; tr->par = l;
l = tr;
t = trr;
}
}
else
{
<T><C>SplayNode* tl = t->lt;
if (tl == 0)
{
++count;
tl = new <T><C>SplayNode(item, def);
comp = 0;
}
else
comp = <T>CMP(item, tl->item);
if (comp >= 0)
{
r->lt = t; t->par = r;
r = t;
t = tl;
}
else
{
<T><C>SplayNode* tll = tl->lt;
if (tll == 0)
{
++count;
tll = new <T><C>SplayNode(item, def);
comp = 0;
}
else
comp = <T>CMP(item, tll->item);
if ((t->lt = tl->rt) != 0) t->lt->par = t;
tl->rt = t; t->par = tl;
r->lt = tl; tl->par = r;
r = tl;
t = tll;
}
}
}
if ((r->lt = t->rt) != 0) r->lt->par = r;
if ((l->rt = t->lt) != 0) l->rt->par = l;
if ((t->lt = dummy->rt) != 0) t->lt->par = t;
if ((t->rt = dummy->lt) != 0) t->rt->par = t;
t->par = 0;
root = t;
return root->cont;
}
void <T><C>SplayMap::del(<T&> key)
{
<T><C>SplayNode* t = (<T><C>SplayNode*)(seek(key));
if (t == 0) return;
<T><C>SplayNode* p = t->par;
--count;
if (t->rt == 0)
{
if (t == root)
{
if ((root = t->lt) != 0) root->par = 0;
}
else if (t == p->lt)
{
if ((p->lt = t->lt) != 0) p->lt->par = p;
}
else
if ((p->rt = t->lt) != 0) p->rt->par = p;
}
else
{
<T><C>SplayNode* r = t->rt;
<T><C>SplayNode* l = r->lt;
for(;;)
{
if (l == 0)
{
if (t == root)
{
root = r;
r->par = 0;
}
else if (t == p->lt)
{
p->lt = r;
r->par = p;
}
else
{
p->rt = r;
r->par = p;
}
if ((r->lt = t->lt) != 0) r->lt->par = r;
break;
}
else
{
if ((r->lt = l->rt) != 0) r->lt->par = r;
l->rt = r; r->par = l;
r = l;
l = l->lt;
}
}
}
delete t;
}
void <T><C>SplayMap::_kill(<T><C>SplayNode* t)
{
if (t != 0)
{
_kill(t->lt);
_kill(t->rt);
delete t;
}
}
<T><C>SplayNode* <T><C>SplayMap::_copy(<T><C>SplayNode* t)
{
if (t != 0)
{
<T><C>SplayNode* l = _copy(t->lt);
<T><C>SplayNode* r = _copy(t->rt);
<T><C>SplayNode* x = new <T><C>SplayNode(t->item, t->cont, l, r);
if (l != 0) l->par = x;
if (r != 0) r->par = x;
return x;
}
else
return 0;
}
int <T><C>SplayMap::OK()
{
int v = 1;
if (root == 0)
v = count == 0;
else
{
int n = 1;
<T><C>SplayNode* trail = leftmost();
<T><C>SplayNode* t = succ(trail);
while (t != 0)
{
++n;
v &= <T>CMP(trail->item, t->item) < 0;
trail = t;
t = succ(t);
}
v &= n == count;
}
if (!v) error("invariant failure");
return v;
}