택배 상자 꺼내기(재도전)

- 문제

 

- 생각

역순으로 쌓이는 상자를 어떻게 처리하지?

 

- 코드

def solution(n, w, num):
    answer = 0
    storage=[]
    # 높이 계산
    h=n//w+1
    # 상자 번호 계산
    x=1
    for i in range(h):
        t=[]
        # 줄대로 계산
        for j in range(w):
            if x<=n:
                t.append(x)
                x+=1
            # n보다 큰 빈공간은 0으로 채우기
            else :
                t.append(0)
        # 정방향
        if i%2==0:
            storage.append(t)
        # 역방향
        else :
            t.reverse()
            storage.append(t)
    # 저장 행    
    for i in range(len(storage)):
        # 저장 열
        for j in range(len(storage[0])):
            if storage[i][j]== num :
                d=i
                while d<h and storage[d][j]:
                    answer+=1
                    d+=1
    return answer