tvl-depot/users/wpcarro/scratch/facebook/interview-cake/cafe-order-checker.py
Vincent Ambo 019f8fd211 subtree(users/wpcarro): docking briefcase at '24f5a642'
git-subtree-dir: users/wpcarro
git-subtree-mainline: 464bbcb15c
git-subtree-split: 24f5a642af
Change-Id: I6105b3762b79126b3488359c95978cadb3efa789
2021-12-14 02:15:47 +03:00

34 lines
798 B
Python

def valid(take_out, dine_in, served):
# edge case
if len(take_out) + len(dine_in) != len(served):
return False
i = 0
j = 0
k = 0
while i < len(take_out) and j < len(dine_in):
if take_out[i] == served[k]:
i += 1
elif dine_in[j] == served[k]:
j += 1
else:
return False
k += 1
# take out
while i < len(take_out):
if take_out[i] != served[k]:
return False
i += 1
# dine in
while j < len(dine_in):
if dine_in[j] != served[k]:
return False
j += 1
return True
take_out = [17, 8, 24]
dine_in = [12, 19, 2]
served = [17, 8, 12, 19, 24, 2]
result = valid(take_out, dine_in, served)
print(result)
assert result
print("Success!")