subtree(users/wpcarro): docking briefcase at '24f5a642'
git-subtree-dir: users/wpcarro git-subtree-mainline:464bbcb15cgit-subtree-split:24f5a642afChange-Id: I6105b3762b79126b3488359c95978cadb3efa789
This commit is contained in:
commit
019f8fd211
766 changed files with 175420 additions and 0 deletions
49
users/wpcarro/scratch/facebook/largest-stack.py
Normal file
49
users/wpcarro/scratch/facebook/largest-stack.py
Normal file
|
|
@ -0,0 +1,49 @@
|
|||
from stack import Stack, from_list
|
||||
from heapq import heapify, heappush, heappop
|
||||
from random import shuffle
|
||||
|
||||
class MaxStack(Stack):
|
||||
def __init__(self):
|
||||
self.max = Stack()
|
||||
super().__init__()
|
||||
|
||||
def __repr__(self):
|
||||
return super().__repr__()
|
||||
|
||||
def push(self, x):
|
||||
super().push(x)
|
||||
max = self.get_max()
|
||||
if not max:
|
||||
self.max.push(x)
|
||||
else:
|
||||
self.max.push(max if x < max else x)
|
||||
|
||||
def pop(self):
|
||||
self.max.pop()
|
||||
return super().pop()
|
||||
|
||||
def get_max(self):
|
||||
return self.max.peek()
|
||||
|
||||
xs = list(range(1, 11))
|
||||
shuffle(xs)
|
||||
stack = MaxStack()
|
||||
for x in xs:
|
||||
stack.push(x)
|
||||
|
||||
print(stack)
|
||||
result = stack.get_max()
|
||||
print(result)
|
||||
assert result == 10
|
||||
|
||||
popped = stack.pop()
|
||||
print("Popped: {}".format(popped))
|
||||
print(stack)
|
||||
while popped != 10:
|
||||
assert stack.get_max() == 10
|
||||
popped = stack.pop()
|
||||
print("Popped: {}".format(popped))
|
||||
print(stack)
|
||||
|
||||
assert stack.get_max() != 10
|
||||
print("Success!")
|
||||
Loading…
Add table
Add a link
Reference in a new issue