题目内容:
洗碗工小明碰上了一位强迫症老板老王,餐厅一共就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()