用python刷mooc算法--强迫症老板和他的洗碗工

806 阅读2分钟

题目内容:

洗碗工小明碰上了一位强迫症老板老王,餐厅一共就10只盘子,老板给仔细编上了0~9等10个号码,并要求小明按照从0到9的编号来洗盘子,当然,每洗好一只盘子,就必须得整齐叠放起来。

小明洗盘子期间,经常就有顾客来取盘子,当然每位顾客只能从盘子堆最上面取1只盘子离开。

老王在收银台仔细地记录了顾客依次取到盘子的编号,比如“1043257689”,这样他就能判断小明是不是遵照命令按照0123456789的次序来洗盘子了。

你也能像老王一样作出准确的判断吗?

输入格式:

长度为10的字符串,其中只包含0~9的数字,且不重复,代表顾客依次取到的盘子编号

输出格式:

字符串:Yes或者No,表示遵照次序洗盘子,或者没有遵照次序洗盘子

输入样例1:

1043257689

输出样例1:

Yes

输入样例2:

4230178956

输出样例2:

No

我的解题思路 遍历取盘子的序号与洗盘子的序号进行比较:

(1)先洗盘子,初始洗盘子栈为空,盘子0先入栈

(2)依次循环比较取盘子的序号是否与洗盘子的栈顶元素一致,如果一致就弹出,不一致,洗盘子序号递增并入栈,全部比较完之后,如果洗盘子的栈为空,结论就是“Yes”,否则就是No

详细代码如下,并已测试通过

class Stack(object):
    def __init__(self):
        self.elem_list = []

    def push(self, elem):
        self.elem_list.append(elem)

    def pop(self):
        return self.elem_list.pop()

    def peek(self):
        return self.elem_list[-1]

    def is_empty(self):
        if len(self.elem_list) == 0:
            return True
        else:
            return False
 def wash_dishes():
    input_str = input().strip()
    str_stack = Stack()
    str_len = len(input_str)
    str_index = 0
    wash_index = 0
    while str_index < str_len and wash_index <= str_len:
        str_elem = int(input_str[str_index])

        if str_stack.is_empty() and wash_index != str_len:
            str_stack.push(wash_index)
            wash_index = wash_index + 1
            continue
        if str_elem == str_stack.peek():
            str_stack.pop()
            str_index = str_index + 1
        else:
            str_stack.push(wash_index)
            wash_index = wash_index + 1
    if str_stack.is_empty():
        print("Yes")
    else:
        print("No")

wash_dishes()