Commit 2336082

author

Amogh Singhal

authored

Create running_median_integers.py

1 parent 98b1ebc commit 2336082Copy full SHA for 2336082

File tree

+30

-0

lines changed

+30

-0

lines changed

Lines changed: 30 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,30 @@`
	`1`	`+# Problem: Find Median from Data Stream`
	`2`	`+`
	`3`	`+def findMedian(stream):`
	`4`	`+ # print(stream)`
	`5`	`+ streamSize = len(stream)`
	`6`	`+`
	`7`	`+ if streamSize == 1:`
	`8`	`+ return stream[0]`
	`9`	`+ else:`
	`10`	`+ stream = sorted(stream)`
	`11`	`+ midPt = streamSize // 2`
	`12`	`+`
	`13`	`+ if streamSize % 2 == 1:`
	`14`	`+ return stream[midPt]`
	`15`	`+ else:`
	`16`	`+ return (stream[midPt] + stream[midPt-1]) // 2`
	`17`	`+`
	`18`	`+def findRunningMedian(inputArray):`
	`19`	`+ medianArray = []`
	`20`	`+`
	`21`	`+ for i in range(0,len(inputArray)):`
	`22`	`+ # print(inputArray)`
	`23`	`+ currentMedian = findMedian(inputArray[0:i+1])`
	`24`	`+ medianArray.append(currentMedian)`
	`25`	`+`
	`26`	`+ return medianArray`
	`27`	`+`
	`28`	`+`
	`29`	`+solution = findRunningMedian([1, 2, 3, 4, 5])`
	`30`	`+print(solution)`

Comments

(0)