【Python】【难度：简单】Leetcode 面试题 03.06. 动物收容所

动物收容所。有家动物收容所只收容狗与猫，且严格遵守“先进先出”的原则。在收养该收容所的动物时，收养人只能收养所有动物中“最老”（由其进入收容所的时间长短而定）的动物，或者可以挑选猫或狗（同时必须收养此类动物中“最老”的）。换言之，收养人不能自由挑选想收养的对象。请创建适用于这个系统的数据结构，实现各种操作方法，比如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%的用户