华为 OD 训练营 · 题解精讲
LC1603. 设计停车系统
LC1603. 设计停车系统 题目描述 请你给一个停车场设计一个停车系统。停车场总共有三种不同大小的车位：大，中和小，每种尺寸分别有固定数目的车位。 请你实现 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 次 解题思路 1、用三个变量分别维护各类车型的剩余空位的数目，每次 addCar(carType) 的时候就把对应类型的停车空位数目 -1。 2、如果当前车型的剩余空位数目 == 0，那么说明无法再停车了，就返回 False；否则说明可以停车，就返回 True。 3、carType 的取值为 1,2,3 ，因此可以使用数组保存停车位的剩余空位数目，根据下标来获取对应车型的停车位剩余数目。 参考代码
参考代码
class ParkingSystem(object):
def __init__(self, big, medium, small):
# 数组的第 0 位置初始化为 0，起到占位符的作用
# 这样在 addCar 方法里面
# park[1] 就是 big 的数量
# park[2] 就是 medium 的数量
# park[3] 就是 small 的数量
self.park = [0, big, medium, small]
def addCar(self, carType):
# 当前类型没有车位了，返回 False
if self.park[carType] == 0:
return False
# 否则当前类型车位数量减少一个
self.park[carType] -= 1
return True