CE_III_38_LAB1/3_plot.py at master · roviso/CE_III_38_LAB1 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
from searchAlgo import LinearSearch
from searchAlgo import BinarySearch
import random
from time import time
import matplotlib.pyplot as plt

n=10000
linearTimeArr =[]
binaryTimeArr=[]
sizeArr =[]

# Linear Search
for i in range(n,n*11,n):
	sizeArr.append(i)
	randomvalues = random.sample(range(i), i)
	startTime = time()
	LinearSearch(randomvalues, randomvalues[i-1])
	endTime = time()
	totalTime = endTime - startTime
	linearTimeArr.append(totalTime)
	print("Linear search for last value in size",i,"is",totalTime)

# Binary Search
for i in range(n,n*11,n):
	randomvalues = random.sample(range(i), i)
	startTime = time()
	randomvalues.sort()
	BinarySearch(randomvalues, randomvalues[i-1])
	endTime = time()
	totalTime = endTime - startTime
	binaryTimeArr.append(totalTime)
	print("Binary search for last value in size",i,"is",totalTime)

# Plot size vs time graph
fig, ax = plt.subplots(1, 1)
ax.plot(sizeArr,linearTimeArr, label = 'Linear Search')
ax.plot(sizeArr,binaryTimeArr, label = 'Binary Search')
legend = ax.legend(loc = 'upper center', fontsize = 'large')
plt.show()