leetcode之设计停车系统（C++）

参考链接

1. https://leetcode-cn.com/problems/design-parking-system/

题目描述

请你给一个停车场设计一个停车系统。停车场总共有三种不同大小的车位：大，中和小，每种尺寸分别有固定数目的车位。

请你实现 ParkingSystem 类：

ParkingSystem(int big, int medium, int small) 初始化 ParkingSystem 类，三个参数分别对应每种停车位的数目。
bool addCar(int carType) 检查是否有 carType 对应的停车位。 carType 有三种类型：大，中，小，分别用数字 1， 2 和 3 表示。一辆车只能停在 carType 对应尺寸的停车位中。如果没有空车位，请返回 false ，否则将该车停入车位并返回 true 。

解题思路

可以使用三个整数分别保存三种车型的剩余个数，每次来车判断一下对应车型是否有车位，有车位则减一，返回true。也可以使用大小为3的数组保存三种不同车型的剩余个数。

代码

使用整数保存

class ParkingSystem {
public:
    int b, m, s;
    ParkingSystem(int big, int medium, int small) : b(big), m(medium), s(small){

    }
    
    bool addCar(int carType) {
        if (carType == 1 && b >= 1)
        {
            b --;
            return true;
        }
        else if(carType == 2 && m >= 1)
        {
            m --;
            return true;
        }
        else if(carType == 3 && s >= 1)
        {
            s --;
            return true;
        }
        return false;
    }
};

使用数组保存

class ParkingSystem {
public:
    vector<int> nums;
    ParkingSystem(int big, int medium, int small) : nums(3) {
        nums[0] = big;
        nums[1] = medium;
        nums[2] = small;
    }
    
    bool addCar(int carType) {
        if (nums[carType - 1] >= 1)
        {
            nums[carType - 1] --;
            return true;
        }
        return false;
    }
};