洗碗工和强迫症老板(python 栈的定义与使用)

洗碗工和强迫症老板(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")
  • 3
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值