文巾解题面试题 03.06. 动物收容所

UQI-LIUWJ

于 2021-05-27 10:37:42 发布

阅读量118

点赞数

分类专栏： leetcode 文章标签： leetcode 数据结构 python 算法

本文链接：https://blog.csdn.net/qq_40206371/article/details/117323425

版权

leetcode 专栏收录该内容

131 篇文章 0 订阅

订阅专栏

本文解析了一个涉及动物分类的AnimalShelf类，介绍了enqueue方法用于添加动物到狗和猫列表，以及dequeueDog和dequeueCat方法用于按类型移除并返回动物。dequeuedAny方法则灵活处理任意类型的动物。

摘要由CSDN通过智能技术生成

1 题目描述

2 解题思路

这题很简单，顺着思路写就ok了

class AnimalShelf:

    def __init__(self):
        self.dog_list=[]
        self.cat_list=[]
        self.total_list=[]

    def enqueue(self, animal: List[int]) -> None:
        animal_num=animal[0]
        animal_type=animal[1]
        self.total_list.append([animal_num,animal_type])
        if(animal_type==0):
            self.cat_list.append([animal_num,animal_type])
        elif(animal_type==1):
            self.dog_list.append([animal_num,animal_type])

    def dequeueAny(self) -> List[int]:
        if(len(self.total_list)==0):
            return([-1,-1])
        ret_num,ret_type=self.total_list.pop(0)
        if(ret_type==0):
            self.cat_list.remove([ret_num,ret_type])
        elif(ret_type==1):
            self.dog_list.remove([ret_num,ret_type])
        return(ret_num,ret_type)

    def dequeueDog(self) -> List[int]:
        if(len(self.dog_list)==0):
            return([-1,-1])
        ret_num,ret_type=self.dog_list.pop(0)
        self.total_list.remove([ret_num,ret_type])
        return(ret_num,ret_type)


    def dequeueCat(self) -> List[int]:
        if(len(self.cat_list)==0):
            return([-1,-1])
        ret_num,ret_type=self.cat_list.pop(0)
        self.total_list.remove([ret_num,ret_type])
        return(ret_num,ret_type)


# 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()