Python; 리스트로 간단한 극장 예약 구현하기

극장의 크기는 10X10이다.

사용자에게 먼저 좌석을 보여준다.

예약된 좌석은 1, 예약이 안된 자리는 0이다.

 

 

def printList(lst):
    for i in range(len(lst)):
        for j in lst[i]:
            print(j,end=' ')
        print()

def isReserved(data,row,col):
    if (data[col][row]==0): return False
    else: return True


seat=[[0 for _ in range(10)] for _ in range(10)]

while True:
    print('--------------------\n1 2 3 4 5 6 7 8 9 10\n--------------------')
    printList(seat)

    row=int(input('원하시는 좌석의 행을 선택하세요 (취소는 -1): '))
    if (row==-1): break
    col=int(input('원하시는 좌석의 열을 선택하세요 (취소는 -1): '))
    if (col==-1): break

    if(not isReserved(seat,row-1,col-1)): 
        seat[col-1][row-1]=1
        print('예약되었습니다.')
    else: print('이미 예약된 좌석입니다. 다시 예매바랍니다.')

 

리스트 함축어를 통해서 0이 10개가 담긴 리스트가 10줄 생성함 (2차원 배열. 10X10)

while True를 통해서 취소하지 않는 한 계속 입력받도록 함

isReserved 변수를 만들어서 좌석 데이터와 좌석 x,y좌표를 매개변수로 입력받아 예약유무 판단