赞
踩
学校的自助午餐提供圆形和方形的三明治,分别用数字0和1表示,所有学生站在一个队列里,每个学生要么喜欢圆形的要么喜欢方形的
餐厅里三明治的数量与学生的数量相同,所有三明治都放在一个栈里,每一轮:
如果队列最前面的学生喜欢栈顶的三明治,那么会拿走它并离开队列。否则,这名学生会放弃这个三明治并返回到队列的尾部。
这个过程会一直持续到队列里所有学生都不喜欢栈顶的三明治为止。
这个过程会一直持续到队列里所有学生都不喜欢栈顶的三明治为止。
给你两个整数数组students 和sandwiches,其中sandwiches[i] 是栈里面第i 个三明治的类型(i=0是栈的顶部),students[j]是初始队列里第j
个三明治的类型(i= 0 是栈的顶部),students[j]是初始队列里第j名学生对三明治的喜好(j=0 是队列的最开始位置)。
请你返回无法吃午餐的学生数量。
披萨的栈顶元素没有被取走,后面的披萨也拿不到。(当前三明治没有被学生喜欢了)
学生的顺序不用考虑,会归队。
代码如下:
解法一:
class Solution1:
def countStudents(self,students,sandwiches) -> int:
counts = {1:0,0:0}
for i in students:
counts[i] +=1
index,san_len= 0,len(sandwiches)
while index < san_len:
if counts[sandwiches[index]] > 0:
counts[sandwiches[index]] -=1
else:
return san_len - index
index +=1
else:
return 0
解法二:
class Solution:
def countStudents(self,students,sandwiches) -> int:
ans = [0,0] # 学生披萨类型的数量
for i in students:
ans[i] +=1
for i in sandwiches:
if ans[i] == 0:
break # 披萨已经不能被学生取走了
ans[i] -= 1
return sum(ans) # 返回剩余的学生的个数
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。