{ "cells": [ { "cell_type": "code", "execution_count": 1, "metadata": { "collapsed": false }, "outputs": [], "source": [ "import numpy as np\n", "from bokeh.plotting import figure, show, output_notebook\n", "from bokeh.io import push_notebook\n", "from time import sleep" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "## algorithm" ] }, { "cell_type": "code", "execution_count": 2, "metadata": { "collapsed": true }, "outputs": [], "source": [ "class Node:\n", " def __init__(self, value, left=None, right=None, red=False):\n", " self.value = value\n", " self.left = left\n", " self.right = right\n", " self.red = red\n", "\n", " @staticmethod\n", " def red(node):\n", " return node and node.red\n", "\n", " def rotate_left(self):\n", " if not Node.red(self.left) and Node.red(self.right):\n", " child = self.right\n", " self.right, child.left = child.left, self\n", " self.red, child.red = True, self.red\n", " return child\n", " else:\n", " return self\n", "\n", " def rotate_right(self):\n", " if Node.red(self.left) and Node.red(self.left.left):\n", " child = self.left\n", " self.left, child.right = child.right, self\n", " self.red, child.red = True, self.red\n", " return child\n", " else:\n", " return self\n", " \n", " def flip_colors(self):\n", " if Node.red(self.left) and Node.red(self.right):\n", " self.red, self.left.red, self.right.red = True, False, False" ] }, { "cell_type": "code", "execution_count": 3, "metadata": { "collapsed": false }, "outputs": [], "source": [ "def insert(node, value, root=True):\n", " if not node:\n", " return Node(value, red=not root)\n", " \n", " # insert value\n", " if value == node.value:\n", " pass\n", " elif value < node.value:\n", " node.left = insert(node.left, value, root=False)\n", " else:\n", " node.right = insert(node.right, value, root=False)\n", " \n", " # update tree w.r.t invariants\n", " node = node.rotate_left()\n", " node = node.rotate_right()\n", " node.flip_colors()\n", "\n", " # keep root black\n", " if root:\n", " node.red = False\n", "\n", " return node" ] }, { "cell_type": "code", "execution_count": 4, "metadata": { "collapsed": true }, "outputs": [], "source": [ "def draw(node, red, black, xlo=0, xhi=1, x=.5, y=0):\n", " if node:\n", " x_, y_ = (xlo + xhi) / 2, y - 1\n", " (red if node.red else black).append([x, y, x_, y_])\n", " draw(node.left, red, black, xlo, x_, x_, y_)\n", " draw(node.right, red, black, x_, xhi, x_, y_)" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "## run" ] }, { "cell_type": "code", "execution_count": 5, "metadata": { "collapsed": false }, "outputs": [], "source": [ "# insert few values\n", "root = None\n", "for i in range(5):\n", " root = insert(root, i)" ] }, { "cell_type": "code", "execution_count": 6, "metadata": { "collapsed": false }, "outputs": [ { "data": { "text/html": [ "\n", "
\n", " \n", " Loading BokehJS ...\n", "
" ] }, "metadata": {}, "output_type": "display_data" }, { "data": { "application/javascript": [ "\n", "(function(global) {\n", " function now() {\n", " return new Date();\n", " }\n", "\n", " var force = true;\n", "\n", " if (typeof (window._bokeh_onload_callbacks) === \"undefined\" || force === true) {\n", " window._bokeh_onload_callbacks = [];\n", " window._bokeh_is_loading = undefined;\n", " }\n", "\n", "\n", " \n", " if (typeof (window._bokeh_timeout) === \"undefined\" || force === true) {\n", " window._bokeh_timeout = Date.now() + 5000;\n", " window._bokeh_failed_load = false;\n", " }\n", "\n", " var NB_LOAD_WARNING = {'data': {'text/html':\n", " \"
\\n\"+\n", " \"

\\n\"+\n", " \"BokehJS does not appear to have successfully loaded. If loading BokehJS from CDN, this \\n\"+\n", " \"may be due to a slow or bad network connection. Possible fixes:\\n\"+\n", " \"

\\n\"+\n", " \"\\n\"+\n", " \"\\n\"+\n", " \"from bokeh.resources import INLINE\\n\"+\n", " \"output_notebook(resources=INLINE)\\n\"+\n", " \"\\n\"+\n", " \"
\"}};\n", "\n", " function display_loaded() {\n", " if (window.Bokeh !== undefined) {\n", " var el = document.getElementById(\"a9a47e6b-6413-432b-b807-c046ca5ab7d7\");\n", " el.textContent = \"BokehJS \" + Bokeh.version + \" successfully loaded.\";\n", " } else if (Date.now() < window._bokeh_timeout) {\n", " setTimeout(display_loaded, 100)\n", " }\n", " }\n", "\n", " function run_callbacks() {\n", " window._bokeh_onload_callbacks.forEach(function(callback) { callback() });\n", " delete window._bokeh_onload_callbacks\n", " console.info(\"Bokeh: all callbacks have finished\");\n", " }\n", "\n", " function load_libs(js_urls, callback) {\n", " window._bokeh_onload_callbacks.push(callback);\n", " if (window._bokeh_is_loading > 0) {\n", " console.log(\"Bokeh: BokehJS is being loaded, scheduling callback at\", now());\n", " return null;\n", " }\n", " if (js_urls == null || js_urls.length === 0) {\n", " run_callbacks();\n", " return null;\n", " }\n", " console.log(\"Bokeh: BokehJS not loaded, scheduling load and callback at\", now());\n", " window._bokeh_is_loading = js_urls.length;\n", " for (var i = 0; i < js_urls.length; i++) {\n", " var url = js_urls[i];\n", " var s = document.createElement('script');\n", " s.src = url;\n", " s.async = false;\n", " s.onreadystatechange = s.onload = function() {\n", " window._bokeh_is_loading--;\n", " if (window._bokeh_is_loading === 0) {\n", " console.log(\"Bokeh: all BokehJS libraries loaded\");\n", " run_callbacks()\n", " }\n", " };\n", " s.onerror = function() {\n", " console.warn(\"failed to load library \" + url);\n", " };\n", " console.log(\"Bokeh: injecting script tag for BokehJS library: \", url);\n", " document.getElementsByTagName(\"head\")[0].appendChild(s);\n", " }\n", " };var element = document.getElementById(\"a9a47e6b-6413-432b-b807-c046ca5ab7d7\");\n", " if (element == null) {\n", " console.log(\"Bokeh: ERROR: autoload.js configured with elementid 'a9a47e6b-6413-432b-b807-c046ca5ab7d7' but no matching script tag was found. \")\n", " return false;\n", " }\n", "\n", " var js_urls = [\"https://cdn.pydata.org/bokeh/release/bokeh-0.12.5.min.js\", \"https://cdn.pydata.org/bokeh/release/bokeh-widgets-0.12.5.min.js\"];\n", "\n", " var inline_js = [\n", " function(Bokeh) {\n", " Bokeh.set_log_level(\"info\");\n", " },\n", " \n", " function(Bokeh) {\n", " \n", " },\n", " \n", " function(Bokeh) {\n", " \n", " document.getElementById(\"a9a47e6b-6413-432b-b807-c046ca5ab7d7\").textContent = \"BokehJS is loading...\";\n", " },\n", " function(Bokeh) {\n", " console.log(\"Bokeh: injecting CSS: https://cdn.pydata.org/bokeh/release/bokeh-0.12.5.min.css\");\n", " Bokeh.embed.inject_css(\"https://cdn.pydata.org/bokeh/release/bokeh-0.12.5.min.css\");\n", " console.log(\"Bokeh: injecting CSS: https://cdn.pydata.org/bokeh/release/bokeh-widgets-0.12.5.min.css\");\n", " Bokeh.embed.inject_css(\"https://cdn.pydata.org/bokeh/release/bokeh-widgets-0.12.5.min.css\");\n", " }\n", " ];\n", "\n", " function run_inline_js() {\n", " \n", " if ((window.Bokeh !== undefined) || (force === true)) {\n", " for (var i = 0; i < inline_js.length; i++) {\n", " inline_js[i](window.Bokeh);\n", " }if (force === true) {\n", " display_loaded();\n", " }} else if (Date.now() < window._bokeh_timeout) {\n", " setTimeout(run_inline_js, 100);\n", " } else if (!window._bokeh_failed_load) {\n", " console.log(\"Bokeh: BokehJS failed to load within specified timeout.\");\n", " window._bokeh_failed_load = true;\n", " } else if (force !== true) {\n", " var cell = $(document.getElementById(\"a9a47e6b-6413-432b-b807-c046ca5ab7d7\")).parents('.cell').data().cell;\n", " cell.output_area.append_execute_result(NB_LOAD_WARNING)\n", " }\n", "\n", " }\n", "\n", " if (window._bokeh_is_loading === 0) {\n", " console.log(\"Bokeh: BokehJS loaded, going straight to plotting\");\n", " run_inline_js();\n", " } else {\n", " load_libs(js_urls, function() {\n", " console.log(\"Bokeh: BokehJS plotting callback run at\", now());\n", " run_inline_js();\n", " });\n", " }\n", "}(this));" ] }, "metadata": {}, "output_type": "display_data" }, { "data": { "text/html": [ "\n", "\n", "
\n", "
\n", "
\n", "" ] }, "metadata": {}, "output_type": "display_data" } ], "source": [ "output_notebook()\n", "\n", "# get plot data\n", "red, black = [], []\n", "draw(root, red, black)\n", "\n", "# plot\n", "plot = figure(plot_height=400, plot_width=800)\n", "plot.axis.visible = False\n", "plot.grid.visible = False\n", "\n", "red_segment = plot.segment(*zip(*red), color='#ff0000')\n", "black_segment = plot.segment(*zip(*black), color='#000000')\n", "\n", "handle = show(plot, notebook_handle=True)" ] }, { "cell_type": "code", "execution_count": 7, "metadata": { "collapsed": false }, "outputs": [], "source": [ "for _ in range(100):\n", " for i in np.random.randint(0, 1000, 5):\n", " root = insert(root, i)\n", "\n", " red, black = [], []\n", " draw(root, red, black)\n", "\n", " x0, y0, x1, y1 = red and zip(*red) or [tuple()] * 4\n", " red_segment.data_source.data.update({\n", " 'x0': x0, 'y0': y0, 'x1': x1, 'y1': y1\n", " })\n", " x0, y0, x1, y1 = black and zip(*black) or [tuple()] * 4\n", " black_segment.data_source.data.update({\n", " 'x0': x0, 'y0': y0, 'x1': x1, 'y1': y1\n", " })\n", "\n", " push_notebook()\n", " sleep(.1)" ] }, { "cell_type": "code", "execution_count": null, "metadata": { "collapsed": true }, "outputs": [], "source": [] } ], "metadata": { "kernelspec": { "display_name": "Python 3", "language": "python", "name": "python3" }, "language_info": { "codemirror_mode": { "name": "ipython", "version": 3 }, "file_extension": ".py", "mimetype": "text/x-python", "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", "version": "3.6.0" } }, "nbformat": 4, "nbformat_minor": 2 }