HYUNGI'S TECH BLOG

과제1) 성수역 지하철 열차 시스템

지하철 노선 클래스

# 2호선 내부순환선 (총 43개 역) 시청역 <->시청역 (내부순환)
# 성수지선 (총 5개 역) 성수역 <-> 신설동
# 신정지선 (총 38개 역) 성수역 <->까치산
# class instance를 만들때 각 노선의 최대 정착역의 수를 구한다.
class SubwayLine:
    def __init__(self, num_of_stn):
        # 지하철역의 수 만큼 [0]리스트를 만들어서
        # 역별로 정차하고 있는 지하철의 수를 관리
        self.num_of_stop_train_list = [0]*num_of_stn

    # 해당 index위치의 count값을 1증가 시켜준다.
    def increaseNumOfStopTrain(self, idx):
        if self.num_of_stop_train_list[idx] == 0:
            self.num_of_stop_train_list[idx] += 1
        else:
            #  위험 알람 method를 호출해서 False값을 반환하도록 한다.
            self.setDangerAlert()

    def decreaseNumOfStopTrain(self, idx):
        if self.num_of_stop_train_list[idx] == 1:
            self.num_of_stop_train_list[idx] -= 1

    # 해당 index위치에 정차되어있는 지하철의 수를 반환한다.
    def getNumOfStopTrain(self, idx):
        return self.num_of_stop_train_list[idx]
    # 해당 역 위치에 지하철이 정차되어있는 경우, False값을 반환해서 출발할 수 없도록 한다.

    def setDangerAlert(self):
        return False