下面是一个用 Python 实现的 ParkingSystem
类:
class ParkingSystem:
def __init__(self, big: int, medium: int, small: int):
# 初始化各类车位的数量
self.slots = {
1: big, # 1 表示大车位
2: medium, # 2 表示中车位
3: small # 3 表示小车位
}
def addCar(self, carType: int) -> bool:
# 检查该类型车位是否还有剩余
if self.slots[carType] > 0:
self.slots[carType] -= 1 # 停一辆车,占用一个车位
return True
else:
return False
使用示例:
parkingSystem = ParkingSystem(1, 1, 0)
print(parkingSystem.addCar(1)) # 返回 True,有一个大车位
print(parkingSystem.addCar(2)) # 返回 True,有一个中车位
print(parkingSystem.addCar(3)) # 返回 False,没有小车位
print(parkingSystem.addCar(1)) # 返回 False,大车位已经用完
这个类结构简单清晰,适合用于各种场景。