洗碗工和强迫症老板(python 栈的定义与使用)
题目:
洗碗工小明碰上了一位强迫症老板老王,餐厅一共就10只盘子,老板给仔细编上了0~9等10个号码,并要求小明按照从0到9的编号来洗盘子,当然,每洗好一只盘子,就必须得整齐叠放起来。
小明洗盘子期间,经常就有顾客来取盘子,当然每位顾客只能从盘子堆最上面取1只盘子离开。老王在收银台仔细地记录了顾客依次取到盘子的编号,比如“1043257689”,这样他就能判断小明是不是遵照命令按照0123456789的次序来洗盘子了。你也能像老王一样作出准确的判断吗?
输入格式:
长度为10的字符串,其中只包含0~9的数字,且不重复,代表顾客依次取到的盘子编号
输出格式:
字符串:Yes或者No,表示遵照次序洗盘子,或者没有遵照次序洗盘子
输入样例1:
1043257689
输出样例1:
Yes
输入样例2:
4230178956
输出样例2:
No
输入样例2:
4321078965
输出样例2:
Yes
class Stack:
def __init__(self, tail_type=True):
self.items = []
self.type = tail_type
def push(self, item):
self.items.append(item) if self.type else self.items.insert(0, item)
def pop(self):
return self.items.pop() if self.type else self.items.pop(0)
def peek(self):
return self.items[-1] if self.type else self.items[0]
def isEmpty(self):
return self.items == []
def size(self):
return len(self.items)
if __name__ == '__main__':
take_order = input()
take_order = [int(plate_no) for plate_no in take_order]
wished_plate = Stack()
pop_order = []
take_flag = 0 # 用于标记拿走盘子的编号在take_order的索引
for plate_no in range(10):
wished_plate.push(plate_no)
if plate_no == take_order[take_flag]:
# 当目前栈顶的盘子编号和顾客拿盘顺序编号相等时才可能允许顾客拿盘子
while not wished_plate.isEmpty() and wished_plate.peek() == take_order[take_flag]:
# 仅当洗好的盘子不为空并且栈顶的盘子编号和目前拿走的盘子编号相等时才出栈(被拿走)
pop_order.append(wished_plate.pop())
take_flag += 1
while not wished_plate.isEmpty():
pop_order.append(wished_plate.pop())
print("Yes" if pop_order == take_order else "No")