1603. 设计停车系统
https://leetcode-cn.com/problems/design-parking-system/
难度简单71收藏分享切换为英文接收动态反馈
请你给一个停车场设计一个停车系统。停车场总共有三种不同大小的车位:大,中和小,每种尺寸分别有固定数目的车位。
请你实现 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 直接模拟停车
class ParkingSystem:
def __init__(self, big: int, medium: int, small: int):
self.big = big
self.medium = medium
self.small = small
def addCar(self, carType: int) -> bool:
if carType == 1:
self.big -= 1
return self.big >= 0
elif carType == 2:
self.medium -= 1
return self.medium >= 0
elif carType == 3:
self.small -= 1
return self.small >= 0
else:
print(f"unkown car Type :{carType}")
return True
pass
# Your ParkingSystem object will be instantiated and called as such:
# obj = ParkingSystem(big, medium, small)
# param_1 = obj.addCar(carType)
解法2
解法1 有一个问题,每次先减1 , 如果对应的车位已经小于0 ,直接返回False 即可, 不用先减1了。
class ParkingSystem:
def __init__(self, big: int, medium: int, small: int):
self.big = big
self.medium = medium
self.small = small
def addCar(self, carType: int) -> bool:
assert carType in {1,2,3},f"unkown car Type :{carType}"
if carType == 1 and self.big > 0:
self.big -= 1
return True
elif carType == 2 and self.medium > 0:
self.medium -= 1
return True
elif carType == 3 and self.small > 0:
self.small -= 1
return True
return False
解法3
每种类型作为一个数组 保存,而不是用一个变量。 这样看起来比较简洁一点。 数组的下标刚好对应每种类型。零号位置 占领一个空位置。
class ParkingSystem(object):
def __init__(self, big, medium, small):
self.park = [None, big, medium, small]
def addCar(self, carType):
assert carType in {1, 2, 3}, f"Unkown car Type:{carType}"
if self.park[carType] == 0:
return False
self.park[carType] -= 1
return True