// 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, 675 Mass Ave, Cambridge, MA 02139, USA. */ #ifdef __GNUG__ #pragma implementation #endif #include #include ".SplayPQ.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 { SplayNode* lt; SplayNode* rt; SplayNode* par; } _dummy_null; /* traversal primitives */ SplayNode* SplayPQ::leftmost() { SplayNode* t = root; if (t != 0) while (t->lt != 0) t = t->lt; return t; } SplayNode* SplayPQ::rightmost() { SplayNode* t = root; if (t != 0) while (t->rt != 0) t = t->rt; return t; } SplayNode* SplayPQ::succ(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; } } } SplayNode* SplayPQ::pred(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 SplayPQ::seek( key) { SplayNode* t = root; if (t == 0) return 0; int comp = CMP(key, t->item); if (comp == 0) return Pix(t); SplayNode* dummy = (SplayNode*)(&_dummy_null); SplayNode* l = dummy; SplayNode* r = dummy; dummy->rt = dummy->lt = dummy->par = 0; while (comp != 0) { if (comp > 0) { SplayNode* tr = t->rt; if (tr == 0) break; else { comp = 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 = CMP(key, t->item); } } } else { SplayNode* tl = t->lt; if (tl == 0) break; else { comp = 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 = 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; } Pix SplayPQ::enq( item) { ++count; SplayNode* newnode = new SplayNode(item); SplayNode* t = root; if (t == 0) { root = newnode; return Pix(root); } int comp = CMP(item, t->item); SplayNode* dummy = (SplayNode*)(&_dummy_null); SplayNode* l = dummy; SplayNode* r = dummy; dummy->rt = dummy->lt = dummy->par = 0; int done = 0; while (!done) { if (comp >= 0) { SplayNode* tr = t->rt; if (tr == 0) { tr = newnode; comp = 0; done = 1; } else comp = CMP(item, tr->item); if (comp <= 0) { l->rt = t; t->par = l; l = t; t = tr; } else { SplayNode* trr = tr->rt; if (trr == 0) { trr = newnode; comp = 0; done = 1; } else comp = 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 { SplayNode* tl = t->lt; if (tl == 0) { tl = newnode; comp = 0; done = 1; } else comp = CMP(item, tl->item); if (comp >= 0) { r->lt = t; t->par = r; r = t; t = tl; } else { SplayNode* tll = tl->lt; if (tll == 0) { tll = newnode; comp = 0; done = 1; } else comp = 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 Pix(root); } void SplayPQ::del(Pix pix) { SplayNode* t = (SplayNode*)pix; if (t == 0) return; 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 { SplayNode* r = t->rt; 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; } & SplayPQ::front() { if (root == 0) error ("min: empty tree\n"); // else { SplayNode* t = root; SplayNode* l = root->lt; for(;;) { if (l == 0) { root = t; root->par = 0; return root->item; } else { if ((t->lt = l->rt) != 0) t->lt->par = t; l->rt = t; t->par = l; t = l; l = l->lt; } } } } void SplayPQ::del_front() { if (root != 0) { --count; SplayNode* t = root; SplayNode* l = root->lt; if (l == 0) { if ((root = t->rt) != 0) root->par = 0; delete t; } else { for(;;) { SplayNode* ll = l->lt; if (ll == 0) { if ((t->lt = l->rt) != 0) t->lt->par = t; delete l; break; } else { SplayNode* lll = ll->lt; if (lll == 0) { if ((l->lt = ll->rt) != 0) l->lt->par = l; delete ll; break; } else { t->lt = ll; ll->par = t; if ((l->lt = ll->rt) != 0) l->lt->par = l; ll->rt = l; l->par = ll; t = ll; l = lll; } } } } } } SplayPQ::deq() { if (root == 0) error("deq: empty tree"); // else { --count; SplayNode* t = root; SplayNode* l = root->lt; if (l == 0) { if ((root = t->rt) != 0) root->par = 0; res = t->item; delete t; return res; } else { for(;;) { SplayNode* ll = l->lt; if (ll == 0) { if ((t->lt = l->rt) != 0) t->lt->par = t; res = l->item; delete l; return res; } else { SplayNode* lll = ll->lt; if (lll == 0) { if ((l->lt = ll->rt) != 0) l->lt->par = l; res = ll->item; delete ll; return res; } else { t->lt = ll; ll->par = t; if ((l->lt = ll->rt) != 0) l->lt->par = l; ll->rt = l; l->par = ll; t = ll; l = lll; } } } } } } void SplayPQ::_kill(SplayNode* t) { if (t != 0) { _kill(t->lt); _kill(t->rt); delete t; } } SplayNode* SplayPQ::_copy(SplayNode* t) { if (t != 0) { SplayNode* l = _copy(t->lt); SplayNode* r = _copy(t->rt); SplayNode* x = new SplayNode(t->item, l, r); if (l != 0) l->par = x; if (r != 0) r->par = x; return x; } else return 0; } int SplayPQ::OK() { int v = 1; if (root == 0) v = count == 0; else { int n = 1; SplayNode* trail = leftmost(); SplayNode* t = succ(trail); while (t != 0) { ++n; v &= CMP(trail->item, t->item) < 0; trail = t; t = succ(t); } v &= n == count; } if (!v) error("invariant failure"); return v; }