LeetCode/cpp/design-parking-system.cpp

#include <cassert>

class ParkingSystem {
    int big;
    int medium;
    int small;

    int &get(int carType) {
        switch (carType) {
        case 1:
            return big;
        case 2:
            return medium;
        case 3:
            return small;
        default:
            assert(false);
        }
    }

  public:
    ParkingSystem(int big, int medium, int small)
        : big(big), medium(medium), small(small) {}

    bool addCar(int carType) {
        auto &space = get(carType);
        if (space <= 0) {
            return false;
        }

        --space;
        return true;
    }
};

/**
 * Your ParkingSystem object will be instantiated and called as such:
 * ParkingSystem* obj = new ParkingSystem(big, medium, small);
 * bool param_1 = obj->addCar(carType);
 */