added heap sort

Author: fazeVaib

.ipynb_checkpoints/SortingVisualization-checkpoint.ipynb

@@ -2,12 +2,12 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": 11,
+   "execution_count": 1,
    "metadata": {},
    "outputs": [],
    "source": [
     "import plotly.graph_objects as go\n",
-    "from sortings import bubbleSort\n",
+    "from sortings import bubbleSort, insertionSort, quickSort, mergeSort, selectionSort, heapSort\n",
     "from generateData import generateData"
    ]
   },
@@ -20,7 +20,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 7,
+   "execution_count": 2,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -29,13 +29,13 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 8,
+   "execution_count": 3,
    "metadata": {},
    "outputs": [
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "f9371b5c9f754630b4394ab341a69486",
+       "model_id": "70e48aa2c9df4b82ac7405c677f7ae18",
        "version_major": 2,
        "version_minor": 0
       },
@@ -62,14 +62,15 @@
     "                   'zeroline':False,\n",
     "                   'showticklabels':False,\n",
     "                   'title':'Number of operations: 0'},\n",
-    "         'plot_bgcolor':'rgba(0,0,0,0)'}\n",
+    "         'plot_bgcolor':'rgba(0,0,0,0)',\n",
+    "         'modbar': False}\n",
     "fig = go.FigureWidget(data, layout)\n",
     "fig"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 9,
+   "execution_count": 4,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -80,11 +81,16 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 10,
+   "execution_count": 5,
    "metadata": {},
    "outputs": [],
    "source": [
-    "bubbleSort.sort(bar, lay, num, colors)"
+    "# bubbleSort.sort(bar, lay, num, colors)\n",
+    "# insertionSort.sort(bar, lay, num, colors)\n",
+    "# quickSort.sort(bar, lay, num, colors)\n",
+    "# mergeSort.sort(bar, lay, num, colors)\n",
+    "# selectionSort.sort(bar, lay, num, colors)\n",
+    "heapSort.sort(bar, lay, num, colors)"
    ]
   }
  ],

SortingVisualization.ipynb

@@ -7,7 +7,7 @@
    "outputs": [],
    "source": [
     "import plotly.graph_objects as go\n",
-    "from sortings import bubbleSort, insertionSort, quickSort, mergeSort, selectionSort\n",
+    "from sortings import bubbleSort, insertionSort, quickSort, mergeSort, selectionSort, heapSort\n",
     "from generateData import generateData"
    ]
   },
@@ -24,7 +24,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "num = 50"
+    "num = 20"
    ]
   },
   {
@@ -35,7 +35,7 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "efc13b755490462eb56c4cd2566e34c5",
+       "model_id": "319c330ac68e4a219c54781450d2ad96",
        "version_major": 2,
        "version_minor": 0
       },
@@ -57,13 +57,16 @@
     "    )]\n",
     "layout = {'yaxis': {'showgrid':False,\n",
     "                    'showticklabels': False,\n",
-    "                   'zeroline' : False},\n",
+    "                   'zeroline' : False,\n",
+    "                   'fixedrange':True},\n",
     "         'xaxis' : {'showgrid' : False,\n",
     "                   'zeroline':False,\n",
     "                   'showticklabels':False,\n",
+    "                    'fixedrange':True,\n",
     "                   'title':'Number of operations: 0'},\n",
     "         'plot_bgcolor':'rgba(0,0,0,0)'}\n",
     "fig = go.FigureWidget(data, layout)\n",
+    "# fig.show(config={'displayModeBar':False})\n",
     "fig"
    ]
   },
@@ -75,20 +78,21 @@
    "source": [
     "bar = fig.data[0]\n",
     "lay = fig.layout\n",
-    "# print(fig)"
+    "# print(fig.data[0].y)"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
+   "execution_count": 6,
    "metadata": {},
    "outputs": [],
    "source": [
     "# bubbleSort.sort(bar, lay, num, colors)\n",
     "# insertionSort.sort(bar, lay, num, colors)\n",
     "# quickSort.sort(bar, lay, num, colors)\n",
     "# mergeSort.sort(bar, lay, num, colors)\n",
-    "selectionSort.sort(bar, lay, num, colors)"
+    "# selectionSort.sort(bar, lay, num, colors)\n",
+    "heapSort.sort(bar, lay, num, colors)"
    ]
   }
  ],

sortings/__pycache__/heapSort.cpython-36.pyc

@@ -0,0 +1,80 @@
+import time
+
+count = 0
+def heapify(arr, n, i, bar, colors, layout):
+    # Find largest among root and children
+    global count
+    count += 1
+    layout.xaxis.title.text = "Number of operations: " + str(count)
+    largest = i
+    l = 2 * i + 1
+    r = 2 * i + 2
+
+    if l < n and arr[i] < arr[l]:
+        colors[largest] = 'lightslategray'
+        colors[l] = 'yellow'
+        bar.marker.color = colors
+        time.sleep(0.1)
+        largest = l
+
+    if r < n and arr[largest] < arr[r]:
+        colors[largest] = 'lightslategray'
+        largest = r
+        colors[r] = 'yellow'
+        bar.marker.color = colors
+        time.sleep(0.1)
+
+    # If root is not largest, swap with largest and continue heapifying
+    if largest != i:
+        colors[largest] = 'red'
+        colors[i] = 'red'
+        bar.marker.color = colors
+        time.sleep(0.1)
+        arr[i], arr[largest] = arr[largest], arr[i]
+        bar.y = arr
+        colors[largest] = 'lightslategray'
+        colors[i] = 'lightslategray'
+        bar.marker.color = colors
+        time.sleep(0.1)
+        heapify(arr, n, largest, bar, colors, layout)
+
+
+def sort(bar, layout, num, colors):
+    '''
+    bar : first attribute of figure data which consists of dictionary that stores all information of plot
+    layout : layout parameter of the figure
+    num : number of elements to sort
+    colors : color array the consists of color info of all elements to sort 
+    '''
+
+    data = list(bar.y)
+    global count
+    # Build max heap
+    for i in range(num, -1, -1):
+        count += 1
+        layout.xaxis.title.text = "Number of operations: " + str(count)
+        heapify(data, num, i, bar, colors, layout)
+        bar.y = data
+
+    for i in range(num-1, 0, -1):
+        # swap
+        count += 1
+        layout.xaxis.title.text = "Number of operations: " + str(count)
+        colors[i] = 'green'
+        colors[0] = 'green'
+        bar.marker.color = colors
+        time.sleep(0.1)
+        data[i], data[0] = data[0], data[i]
+        bar.y = data
+        colors[i] = 'blue'
+        colors[0] = 'lightslategray'
+        bar.marker.color = colors
+
+        #heapify root element
+        heapify(data, i, 0, bar, colors, layout)
+        bar.y = data
+    
+    colors[0] = 'blue'
+    bar.marker.color = colors
+    count = 0
+    # print(data)