forked from NITSkmOS/Algorithms
-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
SkipList.py: Added SkipList Data Structure Implementation
This adds the implementation of Skip List Data Structure in python, this also adds the operation that can be done in a SkipList Data Structure like Insertion, Deletion, Traversal. Closes NITSkmOS#234
- Loading branch information
1 parent
f0e3dbd
commit 80acf5f
Showing
1 changed file
with
122 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,122 @@ | ||
''' | ||
Skip list is a probabilistic data structure that allows efficient search, insertion and removal operations. | ||
It allows fast search within an ordered sequence of elements, O(log(n)) complexity. | ||
Wikipedia Page: https://en.wikipedia.org/wiki/Skip_list | ||
''' | ||
|
||
from random import randint, seed | ||
import time | ||
|
||
|
||
class SkipNode: | ||
''' | ||
A single node in a SkipList data structure, which is similar to a normal linked list node. | ||
''' | ||
def __init__(self, height = 0, elem = None): | ||
self.elem = elem | ||
self.next = [None]*height | ||
|
||
class SkipList: | ||
|
||
def __init__(self): | ||
|
||
# This will set the head node of the list. | ||
self.head = SkipNode() | ||
self.len = 0 | ||
self.maxHeight = 0 | ||
|
||
def __len__(self): | ||
return self.len | ||
|
||
''' | ||
The Search operation takes O(log(n)) time complexity as compared to O(n) in classical linked list. | ||
''' | ||
def find(self, elem, update = None): | ||
if update == None: | ||
update = self.updateList(elem) | ||
if len(update) > 0: | ||
candidate = update[0].next[0] | ||
if candidate != None and candidate.elem == elem: | ||
return candidate | ||
return None | ||
|
||
''' | ||
This function returns a randomized hieght for insertion algorithm. | ||
''' | ||
def randomHeight(self): | ||
height = 1 | ||
while randint(1, 2) != 1: | ||
height += 1 | ||
return height | ||
|
||
''' | ||
This function returns the skip list that is below the element | ||
which is closest and smaller than the element that is required to find. | ||
''' | ||
def updateList(self, elem): | ||
update = [None]*self.maxHeight | ||
x = self.head | ||
for i in reversed(range(self.maxHeight)): | ||
while x.next[i] != None and x.next[i].elem < elem: | ||
x = x.next[i] | ||
update[i] = x | ||
return update | ||
|
||
''' | ||
This function inserts an element in the correct possition. | ||
The insertion operation takes O(n) time complexity | ||
''' | ||
|
||
def insert(self, elem): | ||
node = SkipNode(self.randomHeight(), elem) | ||
self.maxHeight = max(self.maxHeight, len(node.next)) | ||
while len(self.head.next) < len(node.next): | ||
self.head.next.append(None) | ||
update = self.updateList(elem) | ||
if self.find(elem, update) == None: | ||
for i in range(len(node.next)): | ||
node.next[i] = update[i].next[i] | ||
update[i].next[i] = node | ||
self.len += 1 | ||
|
||
''' | ||
This function will remove the specified element from the list. | ||
The deletion operation in Skip List takes O(log(n)) time. | ||
''' | ||
def remove(self, elem): | ||
update = self.updateList(elem) | ||
x = self.find(elem, update) | ||
if x != None: | ||
for i in reversed(range(len(x.next))): | ||
update[i].next[i] = x.next[i] | ||
if self.head.next[i] == None: | ||
self.maxHeight -= 1 | ||
self.len -= 1 | ||
|
||
''' | ||
This function will print the entier Data Structure, level by level (top to bottom) | ||
''' | ||
def printList(self): | ||
for i in range(len(self.head.next)-1, -1, -1): | ||
x = self.head | ||
while x.next[i] != None: | ||
print x.next[i].elem, | ||
x = x.next[i] | ||
print '' | ||
|
||
if __name__ == "__main__": | ||
skipList = SkipList() | ||
|
||
print("Insertion Started") | ||
t0 = time.time() | ||
for i in range(1000000): | ||
skipList.insert(i) | ||
print("Time taken to insert 1 Million elements to the skip list: "+str(time.time() - t0)+" s.") | ||
print("Total Size = "+str(len(skipList))) | ||
# skipList.printList() | ||
|
||
t0 = time.time() | ||
# Skip List takes the maximum time to search the elements in the middle. | ||
print(skipList.find(596536) != None) | ||
print("Time taken to search in a list of size 1 Million: "+str(time.time() - t0)+" s.") |