一、题目内容
请你给一个停车场设计一个停车系统。停车场总共有三种不同大小的车位:大,中和小,每种尺寸分别有固定数目的车位。
请你实现 ParkingSystem 类:
ParkingSystem(int big, int medium, int small) 初始化 ParkingSystem 类,三个参数分别对应每种停车位的数目。
bool addCar(int carType) 检查是否有 carType 对应的停车位。 carType 有三种类型:大,中,小,分别用数字 1, 2 和 3 表示。一辆车只能停在 carType 对应尺寸的停车位中。如果没有空车位,请返回 false ,否则将该车停入车位并返回 true 。
示例 1:
输入:
[“ParkingSystem”, “addCar”, “addCar”, “addCar”, “addCar”]
[[1, 1, 0], [1], [2], [3], [1]]
输出:
[null, true, true, false, false]
解释:
ParkingSystem parkingSystem = new ParkingSystem(1, 1, 0);
parkingSystem.addCar(1); // 返回 true ,因为有 1 个空的大车位
parkingSystem.addCar(2); // 返回 true ,因为有 1 个空的中车位
parkingSystem.addCar(3); // 返回 false ,因为没有空的小车位
parkingSystem.addCar(1); // 返回 false ,因为没有空的大车位,唯一一个大车位已经被占据了
提示:
0 <= big, medium, small <= 1000
carType 取值为 1, 2 或 3
最多会调用 addCar 函数 1000 次
二、解题思路
每个型号的车库有多少开辟多少,一种类型的车入库,对应型号的车库pop,返回True,直到没有地方就返回False。
三、代码
class ParkingSystem:
def __init__(self, big: int, medium: int, small: int):
self.parking = []
self.big = [-1 for _ in range(big)]
self.medium = [-1 for _ in range(medium)]
self.small = [-1 for _ in range(small)]
self.parking.append(self.big)
self.parking.append(self.medium)
self.parking.append(self.small)
def addCar(self, carType: int) -> bool:
if len(self.parking[carType - 1]) != 0:
self.parking[carType - 1].pop()
return True
else:
return False
Your ParkingSystem object will be instantiated and called as such:
obj = ParkingSystem(big, medium, small)
param_1 = obj.addCar(carType)
if name == ‘main’:
commanders = [“ParkingSystem”, “addCar”, “addCar”, “addCar”, “addCar”]
flows = [[1, 1, 0], [1], [2], [3], [1]]
parkingSystem = ParkingSystem(flows[0][0], flows[0][1], flows[0][2])
res = []
for i in range(1, len(flows)):
for j in range(len(flows[i])):
ans = parkingSystem.addCar(flows[i][j])
res.append(ans)
print(ans)