1102 行
27 KiB
JavaScript

// Copyright 2011 David Galles, University of San Francisco. All rights reserved.
//
// Redistribution and use in source and binary forms, with or without modification, are
// permitted provided that the following conditions are met:
//
// 1. Redistributions of source code must retain the above copyright notice, this list of
// conditions and the following disclaimer.
//
// 2. Redistributions in binary form must reproduce the above copyright notice, this list
// of conditions and the following disclaimer in the documentation and/or other materials
// provided with the distribution.
//
// THIS SOFTWARE IS PROVIDED BY David Galles ``AS IS'' AND ANY EXPRESS OR IMPLIED
// WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND
// FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL <COPYRIGHT HOLDER> OR
// CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
// SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
// ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
// NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
// ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
//
// The views and conclusions contained in the software and documentation are those of the
// authors and should not be interpreted as representing official policies, either expressed
// or implied, of the University of San Francisco
// Constants.
SPLAYTREE.LINK_COLOR = "#007700";
SPLAYTREE.HIGHLIGHT_CIRCLE_COLOR = "#007700";
SPLAYTREE.FOREGROUND_COLOR = "#007700";
SPLAYTREE.BACKGROUND_COLOR = "#EEFFEE";
SPLAYTREE.PRINT_COLOR = SPLAYTREE.FOREGROUND_COLOR;
SPLAYTREE.WIDTH_DELTA = 50;
SPLAYTREE.HEIGHT_DELTA = 50;
SPLAYTREE.STARTING_Y = 50;
SPLAYTREE.FIRST_PRINT_POS_X = 50;
SPLAYTREE.PRINT_VERTICAL_GAP = 20;
SPLAYTREE.PRINT_HORIZONTAL_GAP = 50;
function SPLAYTREE(am, w, h)
{
this.init(am, w, h);
}
SPLAYTREE.prototype = new Algorithm();
SPLAYTREE.prototype.constructor = SPLAYTREE;
SPLAYTREE.superclass = Algorithm.prototype;
SPLAYTREE.prototype.init = function(am, w, h)
{
var sc = SPLAYTREE.superclass;
this.startingX = w / 2;
this.first_print_pos_y = h - 2 * SPLAYTREE.PRINT_VERTICAL_GAP;
this.print_max = w - 10;
var fn = sc.init;
fn.call(this,am);
this.addControls();
this.nextIndex = 0;
this.commands = [];
this.cmd("CreateLabel", 0, "", 20, 10, 0);
this.nextIndex = 1;
this.animationManager.StartNewAnimation(this.commands);
this.animationManager.skipForward();
this.animationManager.clearHistory();
}
SPLAYTREE.prototype.addControls = function()
{
this.insertField = addControlToAlgorithmBar("Text", "");
this.insertField.onkeydown = this.returnSubmit(this.insertField, this.insertCallback.bind(this), 4);
this.insertButton = addControlToAlgorithmBar("Button", "添加");
this.insertButton.onclick = this.insertCallback.bind(this);
this.deleteField = addControlToAlgorithmBar("Text", "");
this.deleteField.onkeydown = this.returnSubmit(this.deleteField, this.deleteCallback.bind(this), 4);
this.deleteButton = addControlToAlgorithmBar("Button", "删除");
this.deleteButton.onclick = this.deleteCallback.bind(this);
this.findField = addControlToAlgorithmBar("Text", "");
this.findField.onkeydown = this.returnSubmit(this.findField, this.findCallback.bind(this), 4);
this.findButton = addControlToAlgorithmBar("Button", "搜索");
this.findButton.onclick = this.findCallback.bind(this);
this.printButton = addControlToAlgorithmBar("Button", "打印");
this.printButton.onclick = this.printCallback.bind(this);
}
SPLAYTREE.prototype.reset = function()
{
this.nextIndex = 1;
this.treeRoot = null;
}
SPLAYTREE.prototype.insertCallback = function(event)
{
var insertedValue = this.insertField.value;
// Get text value
insertedValue = this.normalizeNumber(insertedValue, 4);
if (insertedValue != "")
{
// set text value
this.insertField.value = "";
this.implementAction(this.insertElement.bind(this), insertedValue);
}
}
SPLAYTREE.prototype.deleteCallback = function(event)
{
var deletedValue = this.deleteField.value;
if (deletedValue != "")
{
deletedValue = this.normalizeNumber(deletedValue, 4);
this.deleteField.value = "";
this.implementAction(this.deleteElement.bind(this),deletedValue);
}
}
// TODO: This top-down version is broken. Don't use
SPLAYTREE.prototype.splay = function(value)
{
if (this.treeRoot == null)
{
return false;
}
if (this.treeRoot.data == value)
{
return true;
}
if (value < this.treeRoot.data)
{
if (this.treeRoot.left == null)
{
return false;
}
else if (this.treeRoot.left.data == value)
{
this.singleRotateRight(this.treeRoot);
return true;
}
else if (value < this.treeRoot.left.data)
{
if (this.treeRoot.left.left == null)
{
this.singleRotateRight(this.treeRoot);
return this.splay(value);
}
else
{
this.zigZigRight(this.treeRoot);
return this.splay(value);
}
}
else
{
if (this.treeRoot.left.right == null)
{
this.singleRotateRight(this.treeRoot);
return this.splay(value);
}
else
{
this.doubleRotateRight(this.treeRoot);
return this.splay(value);
}
}
}
else
{
if (this.treeRoot.right == null)
{
return false;
}
else if (this.treeRoot.right.data == value)
{
this.singleRotateLeft(this.treeRoot);
return true;
}
else if (value > this.treeRoot.right.data)
{
if (this.treeRoot.right.right == null)
{
this.singleRotateLeft(this.treeRoot);
return this.splay(value);
}
else
{
this.zigZigLeft(this.treeRoot);
return this.splay(value);
}
}
else
{
if (this.treeRoot.right.left == null)
{
this.singleRotateLeft(this.treeRoot);
return this.splay(value);
}
else
{
this.doubleRotateLeft(this.treeRot);
return this.splay(value);
}
}
}
}
SPLAYTREE.prototype.printCallback = function(event)
{
this.implementAction(this.printTree.bind(this),"");
}
SPLAYTREE.prototype.printTree = function(unused)
{
this.commands = [];
if (this.treeRoot != null)
{
this.highlightID = this.nextIndex++;
var firstLabel = this.nextIndex;
this.cmd("CreateHighlightCircle", this.highlightID, SPLAYTREE.HIGHLIGHT_CIRCLE_COLOR, this.treeRoot.x, this.treeRoot.y);
this.xPosOfNextLabel = SPLAYTREE.FIRST_PRINT_POS_X;
this.yPosOfNextLabel = this.first_print_pos_y;
this.printTreeRec(this.treeRoot);
this.cmd("Delete", this.highlightID);
this.cmd("Step")
for (var i = firstLabel; i < this.nextIndex; i++)
{
this.cmd("Delete", i);
}
this.nextIndex = this.highlightID; /// Reuse objects. Not necessary.
}
return this.commands;
}
SPLAYTREE.prototype.printTreeRec = function(tree)
{
this.cmd("Step");
if (tree.left != null)
{
this.cmd("Move", this.highlightID, tree.left.x, tree.left.y);
this.printTreeRec(tree.left);
this.cmd("Move", this.highlightID, tree.x, tree.y);
this.cmd("Step");
}
var nextLabelID = this.nextIndex++;
this.cmd("CreateLabel", nextLabelID, tree.data, tree.x, tree.y);
this.cmd("SetForegroundColor", nextLabelID, SPLAYTREE.PRINT_COLOR);
this.cmd("Move", nextLabelID, this.xPosOfNextLabel, this.yPosOfNextLabel);
this.cmd("Step");
this.xPosOfNextLabel += SPLAYTREE.PRINT_HORIZONTAL_GAP;
if (this.xPosOfNextLabel > this.print_max)
{
this.xPosOfNextLabel = SPLAYTREE.FIRST_PRINT_POS_X;
this.yPosOfNextLabel += SPLAYTREE.PRINT_VERTICAL_GAP;
}
if (tree.right != null)
{
this.cmd("Move", this.highlightID, tree.right.x, tree.right.y);
this.printTreeRec(tree.right);
this.cmd("Move", this.highlightID, tree.x, tree.y);
this.cmd("Step");
}
return;
}
SPLAYTREE.prototype.findCallback = function(event)
{
var findValue;
findValue = this.normalizeNumber(this.findField.value, 4);
this.findField.value = "";
this.implementAction(this.findElement.bind(this),findValue);
}
SPLAYTREE.prototype.findElement = function(findValue)
{
this.commands = [];
this.highlightID = this.nextIndex++;
var found = this.doFind(this.treeRoot, findValue);
if (found)
{
this.cmd("SetText", 0, "Element " + findValue + " found.");
}
else
{
this.cmd("SetText", 0, "Element " + findValue + " 未找到.");
}
return this.commands;
}
SPLAYTREE.prototype.doFind = function(tree, value)
{
this.cmd("SetText", 0, "正在搜索 "+value);
if (tree != null)
{
this.cmd("SetHighlight", tree.graphicID, 1);
if (tree.data == value)
{
this.cmd("SetText", 0, "正在搜索 "+value+":" + value + " = " + value + " (Element found!)");
this.cmd("Step");
this.cmd("SetText", 0, "Splaying found node to root of tree");
this.cmd("Step");
this.cmd("SetHighlight", tree.graphicID, 0);
this.splayUp(tree);
return true;
}
else
{
if (tree.data > value)
{
this.cmd("SetText", 0, "正在搜索 "+value+":" + value + " < " + tree.data + " (访问左子树)");
this.cmd("Step");
this.cmd("SetHighlight", tree.graphicID, 0);
if (tree.left!= null)
{
this.cmd("CreateHighlightCircle", this.highlightID, SPLAYTREE.HIGHLIGHT_CIRCLE_COLOR, tree.x, tree.y);
this.cmd("Move", this.highlightID, tree.left.x, tree.left.y);
this.cmd("Step");
this.cmd("Delete", this.highlightID);
return this.doFind(tree.left, value);
}
else
{
this.splayUp(tree);
return false;
}
}
else
{
this.cmd("SetText", 0, "正在搜索 "+value+":" + value + " > " + tree.data + " (访问右子树)");
this.cmd("Step");
this.cmd("SetHighlight", tree.graphicID, 0);
if (tree.right!= null)
{
this.cmd("CreateHighlightCircle", this.highlightID, SPLAYTREE.HIGHLIGHT_CIRCLE_COLOR, tree.x, tree.y);
this.cmd("Move", this.highlightID, tree.right.x, tree.right.y);
this.cmd("Step");
this.cmd("Delete", this.highlightID);
return this.doFind(tree.right, value);
}
else
{
this.splayUp(tree);
return false;
}
}
}
}
else
{
this.cmd("SetText", 0, "正在搜索 "+value+":" + "< 空树 > (找不到元素)");
this.cmd("Step");
this.cmd("SetText", 0, "正在搜索 "+value+":" + " (找不到元素)");
return false;
}
}
SPLAYTREE.prototype.insertElement = function(insertedValue)
{
this.commands = new Array();
this.cmd("SetText", 0, "Inserting "+insertedValue);
this.highlightID = this.nextIndex++;
if (this.treeRoot == null)
{
this.cmd("CreateCircle", this.nextIndex, insertedValue, this.startingX, SPLAYTREE.STARTING_Y);
this.cmd("SetForegroundColor", this.nextIndex, SPLAYTREE.FOREGROUND_COLOR);
this.cmd("SetBackgroundColor", this.nextIndex, SPLAYTREE.BACKGROUND_COLOR);
this.cmd("Step");
this.treeRoot = new BSTNode(insertedValue, this.nextIndex, this.startingX, SPLAYTREE.STARTING_Y)
this.nextIndex += 1;
}
else
{
this.cmd("CreateCircle", this.nextIndex, insertedValue, 100, 100);
this.cmd("SetForegroundColor", this.nextIndex, SPLAYTREE.FOREGROUND_COLOR);
this.cmd("SetBackgroundColor", this.nextIndex, SPLAYTREE.BACKGROUND_COLOR);
this.cmd("Step");
var insertElem = new BSTNode(insertedValue, this.nextIndex, 100, 100)
this.nextIndex += 1;
this.cmd("SetHighlight", insertElem.graphicID, 1);
this.insert(insertElem, this.treeRoot)
this.resizeTree();
this.cmd("SetText", 0 , "Splay inserted element to root of tree");
this.cmd("Step");
this.splayUp(insertElem);
}
this.cmd("SetText", 0, "");
return this.commands;
}
SPLAYTREE.prototype.insert = function(elem, tree)
{
this.cmd("SetHighlight", tree.graphicID , 1);
this.cmd("SetHighlight", elem.graphicID , 1);
if (elem.data < tree.data)
{
this.cmd("SetText", 0, elem.data + " < " + tree.data + ", 访问左子树");
}
else
{
this.cmd("SetText", 0, elem.data + " >= " + tree.data + ", 访问右子树");
}
this.cmd("Step");
this.cmd("SetHighlight", tree.graphicID, 0);
this.cmd("SetHighlight", elem.graphicID, 0);
if (elem.data < tree.data)
{
if (tree.left == null)
{
this.cmd("SetText", 0,"Found null tree, inserting element");
this.cmd("SetHighlight", elem.graphicID, 0);
tree.left=elem;
elem.parent = tree;
this.cmd("Connect", tree.graphicID, elem.graphicID, SPLAYTREE.LINK_COLOR);
}
else
{
this.cmd("CreateHighlightCircle", this.highlightID, SPLAYTREE.HIGHLIGHT_CIRCLE_COLOR, tree.x, tree.y);
this.cmd("Move", this.highlightID, tree.left.x, tree.left.y);
this.cmd("Step");
this.cmd("Delete", this.highlightID);
this.insert(elem, tree.left);
}
}
else
{
if (tree.right == null)
{
this.cmd("SetText", 0, "Found null tree, inserting element");
this.cmd("SetHighlight", elem.graphicID, 0);
tree.right=elem;
elem.parent = tree;
this.cmd("Connect", tree.graphicID, elem.graphicID, SPLAYTREE.LINK_COLOR);
elem.x = tree.x + SPLAYTREE.WIDTH_DELTA/2;
elem.y = tree.y + SPLAYTREE.HEIGHT_DELTA
this.cmd("Move", elem.graphicID, elem.x, elem.y);
}
else
{
this.cmd("CreateHighlightCircle", this.highlightID, SPLAYTREE.HIGHLIGHT_CIRCLE_COLOR, tree.x, tree.y);
this.cmd("Move", this.highlightID, tree.right.x, tree.right.y);
this.cmd("Step");
this.cmd("Delete", this.highlightID);
this.insert(elem, tree.right);
}
}
}
SPLAYTREE.prototype.deleteElement = function(deletedValue)
{
this.commands = [];
this.cmd("SetText", 0, "正在刪除 "+deletedValue);
this.cmd("Step");
this.cmd("SetText", 0, "");
this.highlightID = this.nextIndex++;
this.treeDelete(this.treeRoot, deletedValue);
this.cmd("SetText", 0, "");
// Do delete
return this.commands;
}
SPLAYTREE.prototype.treeDelete = function(tree, valueToDelete)
{
this.cmd("SetText", 0, "Finding "+valueToDelete + " and splaying to rooot");
this.cmd("Step");
var inTree = this.doFind(this.treeRoot, valueToDelete);
this.cmd("SetText", 0, "Removing root, leaving left and right trees");
this.cmd("Step")
if (inTree)
{
if (this.treeRoot.right == null)
{
this.cmd("Delete", this.treeRoot.graphicID);
this.cmd("SetText", 0, "No right tree, make left tree the root.");
this.cmd("Step");
this.treeRoot = this.treeRoot.left;
this.treeRoot.parent = null;
this.resizeTree();
}
else if (this.treeRoot.left == null)
{
this.cmd("Delete", this.treeRoot.graphicID);
this.cmd("SetText", 0, "No left tree, make right tree the root.");
this.cmd("Step");
this.treeRoot = this.treeRoot.right
this.treeRoot.parent = null;
this.resizeTree();
}
else
{
var right = this.treeRoot.right;
var left = this.treeRoot.left;
var oldGraphicID = this.treeRoot.graphicID;
this.cmd("Disconnect", this.treeRoot.graphicID, left.graphicID);
this.cmd("Disconnect", this.treeRoot.graphicID, right.graphicID);
this.cmd("SetAlpha", this.treeRoot.graphicID, 0);
this.cmd("SetText", 0, "Splay largest element in left tree to root");
this.cmd("Step")
left.parent = null;
var largestLeft = this.findMax(left);
this.splayUp(largestLeft);
this.cmd("SetText", 0, "Left tree now has no right subtree, connect left and right trees");
this.cmd("Step");
this.cmd("Connect", largestLeft.graphicID, right.graphicID, SPLAYTREE.LINK_COLOR);
largestLeft.parent = null;
largestLeft.right = right;
right.parent = largestLeft;
this.treeRoot = largestLeft;
this.cmd("Delete", oldGraphicID);
this.resizeTree();
}
}
}
SPLAYTREE.prototype.singleRotateRight = function(tree)
{
var B = tree;
var t3 = B.right;
var A = tree.left;
var t1 = A.left;
var t2 = A.right;
this.cmd("SetText", 0, "Zig Right");
this.cmd("SetEdgeHighlight", B.graphicID, A.graphicID, 1);
this.cmd("Step");
if (t2 != null)
{
this.cmd("Disconnect", A.graphicID, t2.graphicID);
this.cmd("Connect", B.graphicID, t2.graphicID, SPLAYTREE.LINK_COLOR);
t2.parent = B;
}
this.cmd("Disconnect", B.graphicID, A.graphicID);
this.cmd("Connect", A.graphicID, B.graphicID, SPLAYTREE.LINK_COLOR);
A.parent = B.parent;
if (B.parent == null)
{
this.treeRoot = A;
}
else
{
this.cmd("Disconnect", B.parent.graphicID, B.graphicID, SPLAYTREE.LINK_COLOR);
this.cmd("Connect", B.parent.graphicID, A.graphicID, SPLAYTREE.LINK_COLOR)
if (B.isLeftChild())
{
B.parent.left = A;
}
else
{
B.parent.right = A;
}
}
A.right = B;
B.parent = A;
B.left = t2;
this.resizeTree();
}
SPLAYTREE.prototype.zigZigRight = function(tree)
{
var C = tree;
var B = tree.left;
var A = tree.left.left;
var t1 = A.left;
var t2 = A.right;
var t3 = B.right;
var t4 = C.right;
this.cmd("SetText", 0, "Zig-Zig Right");
this.cmd("SetEdgeHighlight", C.graphicID, B.graphicID, 1);
this.cmd("SetEdgeHighlight", B.graphicID, A.graphicID, 1);
this.cmd("Step");
this.cmd("SetEdgeHighlight", C.graphicID, B.graphicID, 0);
this.cmd("SetEdgeHighlight", B.graphicID, A.graphicID, 0);
if (C.parent != null)
{
this.cmd("Disconnect", C.parent.graphicID, C.graphicID);
this.cmd("Connect", C.parent.graphicID, A.graphicID, SPLAYTREE.LINK_COLOR);
if (C.isLeftChild())
{
C.parent.left = A;
}
else
{
C.parent.right = A;
}
}
else
{
this.treeRoot = A;
}
if (t2 != null)
{
this.cmd("Disconnect", A.graphicID, t2.graphicID);
this.cmd("Connect", B.graphicID, t2.graphicID, SPLAYTREE.LINK_COLOR);
t2.parent = B;
}
if (t3 != null)
{
this.cmd("Disconnect", B.graphicID, t3.graphicID);
this.cmd("Connect", C.graphicID, t3.graphicID, SPLAYTREE.LINK_COLOR);
t3.parent = C;
}
this.cmd("Disconnect", B.graphicID, A.graphicID);
this.cmd("Connect", A.graphicID, B.graphicID, SPLAYTREE.LINK_COLOR);
this.cmd("Disconnect", C.graphicID, B.graphicID);
this.cmd("Connect", B.graphicID, C.graphicID, SPLAYTREE.LINK_COLOR);
A.right = B;
A.parent = C.parent;
B.parent = A;
B.left = t2;
B.right = C;
C.parent = B;
C.left = t3;
this.resizeTree();
}
SPLAYTREE.prototype.zigZigLeft = function(tree)
{
var A = tree;
var B = tree.right;
var C = tree.right.right;
var t1 = A.left;
var t2 = B.left;
var t3 = C.left;
var t4 = C.right;
this.cmd("SetText", 0, "Zig-Zig Left");
this.cmd("SetEdgeHighlight", A.graphicID, B.graphicID, 1);
this.cmd("SetEdgeHighlight", B.graphicID, C.graphicID, 1);
this.cmd("Step");
this.cmd("SetEdgeHighlight", A.graphicID, B.graphicID, 0);
this.cmd("SetEdgeHighlight", B.graphicID, C.graphicID, 0);
if (A.parent != null)
{
this.cmd("Disconnect", A.parent.graphicID, A.graphicID);
this.cmd("Connect", A.parent.graphicID, C.graphicID, SPLAYTREE.LINK_COLOR);
if (A.isLeftChild())
{
A.parent.left = C;
}
else
{
A.parent.right = C;
}
}
else
{
this.treeRoot = C;
}
if (t2 != null)
{
this.cmd("Disconnect", B.graphicID, t2.graphicID);
this.cmd("Connect", A.graphicID, t2.graphicID, SPLAYTREE.LINK_COLOR);
t2.parent = A;
}
if (t3 != null)
{
this.cmd("Disconnect", C.graphicID, t3.graphicID);
this.cmd("Connect", B.graphicID, t3.graphicID, SPLAYTREE.LINK_COLOR);
t3.parent = B;
}
this.cmd("Disconnect", A.graphicID, B.graphicID);
this.cmd("Disconnect", B.graphicID, C.graphicID);
this.cmd("Connect", C.graphicID, B.graphicID, SPLAYTREE.LINK_COLOR);
this.cmd("Connect", B.graphicID, A.graphicID, SPLAYTREE.LINK_COLOR);
C.parent = A.parent;
A.right = t2;
B.left = A;
A.parent = B;
B.right = t3;
C.left = B;
B.parent = C;
this.resizeTree();
}
SPLAYTREE.prototype.singleRotateLeft = function(tree)
{
var A = tree;
var B = tree.right;
var t1 = A.left;
var t2 = B.left;
var t3 = B.right;
this.cmd("SetText", 0, "Zig Left");
this.cmd("SetEdgeHighlight", A.graphicID, B.graphicID, 1);
this.cmd("Step");
if (t2 != null)
{
this.cmd("Disconnect", B.graphicID, t2.graphicID);
this.cmd("Connect", A.graphicID, t2.graphicID, SPLAYTREE.LINK_COLOR);
t2.parent = A;
}
this.cmd("Disconnect", A.graphicID, B.graphicID);
this.cmd("Connect", B.graphicID, A.graphicID, SPLAYTREE.LINK_COLOR);
B.parent = A.parent;
if (A.parent == null)
{
this.treeRoot = B;
}
else
{
this.cmd("Disconnect", A.parent.graphicID, A.graphicID, SPLAYTREE.LINK_COLOR);
this.cmd("Connect", A.parent.graphicID, B.graphicID, SPLAYTREE.LINK_COLOR)
if (A.isLeftChild())
{
A.parent.left = B;
}
else
{
A.parent.right = B;
}
}
B.left = A;
A.parent = B;
A.right = t2;
this.resizeTree();
}
SPLAYTREE.prototype.splayUp = function(tree)
{
if (tree.parent == null)
{
return;
}
else if (tree.parent.parent == null)
{
if (tree.isLeftChild())
{
this.singleRotateRight(tree.parent);
}
else
{
this.singleRotateLeft(tree.parent);
}
}
else if (tree.isLeftChild() && !tree.parent.isLeftChild())
{
this.doubleRotateLeft(tree.parent.parent);
this.splayUp(tree);
}
else if (!tree.isLeftChild() && tree.parent.isLeftChild())
{
this.doubleRotateRight(tree.parent.parent);
this.splayUp(tree);
}
else if (tree.isLeftChild())
{
this.zigZigRight(tree.parent.parent);
this.splayUp(tree);
}
else
{
this.zigZigLeft(tree.parent.parent);
this.splayUp(tree);
}
}
SPLAYTREE.prototype.findMax = function(tree)
{
if (tree.right != null)
{
this.highlightID = this.nextIndex++;
this.cmd("CreateHighlightCircle", this.highlightID, SPLAYTREE.HIGHLIGHT_CIRCLE_COLOR, tree.x, tree.y);
this.cmd("Step");
while(tree.right != null)
{
this.cmd("Move", this.highlightID, tree.right.x, tree.right.y);
this.cmd("Step");
tree = tree.right;
}
this.cmd("Delete", this.highlightID);
return tree;
}
else
{
return tree;
}
}
SPLAYTREE.prototype.doubleRotateRight = function(tree)
{
this.cmd("SetText", 0, "Zig-Zag Right");
var A = tree.left;
var B = tree.left.right;
var C = tree;
var t1 = A.left;
var t2 = B.left;
var t3 = B.right;
var t4 = C.right;
this.cmd("SetEdgeHighlight", C.graphicID, A.graphicID, 1);
this.cmd("SetEdgeHighlight", A.graphicID, B.graphicID, 1);
this.cmd("Step");
if (t2 != null)
{
this.cmd("Disconnect",B.graphicID, t2.graphicID);
t2.parent = A;
A.right = t2;
this.cmd("Connect", A.graphicID, t2.graphicID, SPLAYTREE.LINK_COLOR);
}
if (t3 != null)
{
this.cmd("Disconnect",B.graphicID, t3.graphicID);
t3.parent = C;
C.left = t2;
this.cmd("Connect", C.graphicID, t3.graphicID, SPLAYTREE.LINK_COLOR);
}
if (C.parent == null)
{
B.parent = null;
this.treeRoot = B;
}
else
{
this.cmd("Disconnect",C.parent.graphicID, C.graphicID);
this.cmd("Connect",C.parent.graphicID, B.graphicID, SPLAYTREE.LINK_COLOR);
if (C.isLeftChild())
{
C.parent.left = B
}
else
{
C.parent.right = B;
}
B.parent = C.parent;
C.parent = B;
}
this.cmd("Disconnect", C.graphicID, A.graphicID);
this.cmd("Disconnect", A.graphicID, B.graphicID);
this.cmd("Connect", B.graphicID, A.graphicID, SPLAYTREE.LINK_COLOR);
this.cmd("Connect", B.graphicID, C.graphicID, SPLAYTREE.LINK_COLOR);
B.left = A;
A.parent = B;
B.right=C;
C.parent=B;
A.right=t2;
C.left = t3;
this.resizeTree();
}
SPLAYTREE.prototype.doubleRotateLeft = function(tree)
{
this.cmd("SetText", 0, "Zig-Zag Left");
var A = tree;
var B = tree.right.left;
var C = tree.right;
var t1 = A.left;
var t2 = B.left;
var t3 = B.right;
var t4 = C.right;
this.cmd("SetEdgeHighlight", A.graphicID, C.graphicID, 1);
this.cmd("SetEdgeHighlight", C.graphicID, B.graphicID, 1);
this.cmd("Step");
if (t2 != null)
{
this.cmd("Disconnect",B.graphicID, t2.graphicID);
t2.parent = A;
A.right = t2;
this.cmd("Connect", A.graphicID, t2.graphicID, SPLAYTREE.LINK_COLOR);
}
if (t3 != null)
{
this.cmd("Disconnect",B.graphicID, t3.graphicID);
t3.parent = C;
C.left = t2;
this.cmd("Connect", C.graphicID, t3.graphicID, SPLAYTREE.LINK_COLOR);
}
if (A.parent == null)
{
B.parent = null;
this.treeRoot = B;
}
else
{
this.cmd("Disconnect",A.parent.graphicID, A.graphicID);
this.cmd("Connect",A.parent.graphicID, B.graphicID, SPLAYTREE.LINK_COLOR);
if (A.isLeftChild())
{
A.parent.left = B
}
else
{
A.parent.right = B;
}
B.parent = A.parent;
A.parent = B;
}
this.cmd("Disconnect", A.graphicID, C.graphicID);
this.cmd("Disconnect", C.graphicID, B.graphicID);
this.cmd("Connect", B.graphicID, A.graphicID, SPLAYTREE.LINK_COLOR);
this.cmd("Connect", B.graphicID, C.graphicID, SPLAYTREE.LINK_COLOR);
B.left = A;
A.parent = B;
B.right=C;
C.parent=B;
A.right=t2;
C.left = t3;
this.resizeTree();
}
SPLAYTREE.prototype.resizeTree = function()
{
var startingPoint = this.startingX;
this.resizeWidths(this.treeRoot);
if (this.treeRoot != null)
{
if (this.treeRoot.leftWidth > startingPoint)
{
startingPoint = this.treeRoot.leftWidth;
}
else if (this.treeRoot.rightWidth > startingPoint)
{
startingPoint = Math.max(this.treeRoot.leftWidth, 2 * startingPoint - this.treeRoot.rightWidth);
}
this.setNewPositions(this.treeRoot, startingPoint, SPLAYTREE.STARTING_Y, 0);
this.animateNewPositions(this.treeRoot);
this.cmd("Step");
}
}
SPLAYTREE.prototype.setNewPositions = function(tree, xPosition, yPosition, side)
{
if (tree != null)
{
tree.y = yPosition;
if (side == -1)
{
xPosition = xPosition - tree.rightWidth;
}
else if (side == 1)
{
xPosition = xPosition + tree.leftWidth;
}
tree.x = xPosition;
this.setNewPositions(tree.left, xPosition, yPosition + SPLAYTREE.HEIGHT_DELTA, -1)
this.setNewPositions(tree.right, xPosition, yPosition + SPLAYTREE.HEIGHT_DELTA, 1)
}
}
SPLAYTREE.prototype.animateNewPositions = function(tree)
{
if (tree != null)
{
this.cmd("Move", tree.graphicID, tree.x, tree.y);
this.animateNewPositions(tree.left);
this.animateNewPositions(tree.right);
}
}
SPLAYTREE.prototype.resizeWidths = function(tree)
{
if (tree == null)
{
return 0;
}
tree.leftWidth = Math.max(this.resizeWidths(tree.left), SPLAYTREE.WIDTH_DELTA / 2);
tree.rightWidth = Math.max(this.resizeWidths(tree.right), SPLAYTREE.WIDTH_DELTA / 2);
return tree.leftWidth + tree.rightWidth;
}
function BSTNode(val, id, initialX, initialY)
{
this.data = val;
this.x = initialX;
this.y = initialY;
this.graphicID = id;
this.left = null;
this.right = null;
this.parent = null;
}
BSTNode.prototype.isLeftChild = function()
{
if (this. parent == null)
{
return true;
}
return this.parent.left == this;
}
SPLAYTREE.prototype.disableUI = function(event)
{
this.insertField.disabled = true;
this.insertButton.disabled = true;
this.deleteField.disabled = true;
this.deleteButton.disabled = true;
this.findField.disabled = true;
this.findButton.disabled = true;
this.printButton.disabled = true;
}
SPLAYTREE.prototype.enableUI = function(event)
{
this.insertField.disabled = false;
this.insertButton.disabled = false;
this.deleteField.disabled = false;
this.deleteButton.disabled = false;
this.findField.disabled = false;
this.findButton.disabled = false;
this.printButton.disabled = false;
}
var currentAlg;
function init()
{
var animManag = initCanvas();
currentAlg = new SPLAYTREE(animManag, canvas.width, canvas.height);
}