动物收容所。有家动物收容所只收容狗与猫,且严格遵守“先进先出”的原则。在收养该收容所的动物时,收养人只能收养所有动物中“最老”(由其进入收容所的时间长短而定)的动物,或者可以挑选猫或狗(同时必须收养此类动物中“最老”的)。换言之,收养人不能自由挑选想收养的对象。请创建适用于这个系统的数据结构,实现各种操作方法,比如enqueue、dequeueAny、dequeueDog和dequeueCat。允许使用Java内置的LinkedList数据结构。
enqueue方法有一个animal参数,animal[0]代表动物编号,animal[1]代表动物种类,其中 0 代表猫,1 代表狗。
dequeue*方法返回一个列表[动物编号, 动物种类],若没有可以收养的动物,则返回[-1,-1]。
示例1:
输入:
["AnimalShelf", "enqueue", "enqueue", "dequeueCat", "dequeueDog", "dequeueAny"]
[[], [[0, 0]], [[1, 0]], [], [], []]
输出:
[null,null,null,[0,0],[-1,-1],[1,0]]
示例2:
输入:
["AnimalShelf", "enqueue", "enqueue", "enqueue", "dequeueDog", "dequeueCat", "dequeueAny"]
[[], [[0, 0]], [[1, 0]], [[2, 1]], [], [], []]
输出:
[null,null,null,null,[2,1],[0,0],[1,0]]
说明:
收纳所的最大容量为20000
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/animal-shelter-lcci
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class AnimalShelf(object):
def __init__(self):
self.q1=[]
self.p1=[]
self.q2=[]
self.p2=[]
self.cnt=0
def enqueue(self, animal):
"""
:type animal: List[int]
:rtype: None
"""
self.cnt+=1
if animal[1]==0:
self.q1.append(animal[0])
self.p1.append(self.cnt)
else:
self.q2.append(animal[0])
self.p2.append(self.cnt)
def dequeueAny(self):
"""
:rtype: List[int]
"""
if not len(self.q1) and not len(self.q2):
return [-1,-1]
if not len(self.q1):
self.p2.pop(0)
return [self.q2.pop(0),1]
if not len(self.q2):
self.p1.pop(0)
return [self.q1.pop(0),0]
if self.p1[0]<self.p2[0]:
self.p1.pop(0)
return [self.q1.pop(0),0]
else:
self.p2.pop(0)
return [self.q2.pop(0),1]
def dequeueDog(self):
"""
:rtype: List[int]
"""
if len(self.q2):
self.p2.pop(0)
return [self.q2.pop(0),1]
else:
return [-1,-1]
def dequeueCat(self):
"""
:rtype: List[int]
"""
if len(self.q1):
self.p1.pop(0)
return [self.q1.pop(0),0]
else:
return [-1,-1]
# Your AnimalShelf object will be instantiated and called as such:
# obj = AnimalShelf()
# obj.enqueue(animal)
# param_2 = obj.dequeueAny()
# param_3 = obj.dequeueDog()
# param_4 = obj.dequeueCat()
执行结果:
通过
显示详情
执行用时 :376 ms, 在所有 Python 提交中击败了70.67%的用户
内存消耗 :22 MB, 在所有 Python 提交中击败了100.00%的用户