물건 사고 파는 문제 파이썬 order, sell, return

입력이 아래처럼, sell, order, return 으로 들어왔을때, 물건을 저장하고 팔 수있다. 

'order' 주문했을때 물건의 이름, 갯수, 가격을 받는다. 

'sell' 명령어를 받았을 때는 해당 아이템 이름을 가진 재고중, 가장 저렴한 아이템부터 판매한다.

'return' 명령어를 받았다면, 해당 아이템을 입력받은 갯수와 가격으로 저장한다. 

 

파이썬 딕셔너리를 이용해서 풀었는데, 

코드가 조금 깨끗하지 않다, 판매할때마다 딕셔너리 내의 밸류 기준으로 정렬했다. 

input = [
    ["order", "item1", "3", "400"],
    ["order", "item2", "3", "700"],
    ["sell", "item1", "1"],
    ["return", "item1", "1", "200"],
    ["sell", "item2", "3"],
    ["return", "item2", "1", "350"],
    ["sell", "item1", "1"],
    ["return", "item1", "1", "100"],
    ["sell", "item1", "2"],
]

def solution(ip):
    ans = []
    product = dict()
    lenInput = len(input)
    for i in range(lenInput):
        # print(product)
        if ip[i][0] == "order":
            product[ip[i][1]] = [[int(ip[i][3]), int(ip[i][2])]]
        elif ip[i][0] == "sell":
            product[ip[i][1]] = sorted(product[ip[i][1]])
            # print(product)
            if ip[i][1] in product:
                tempCnt = int(ip[i][2])
                while tempCnt:
                    if product[ip[i][1]][0][1] > tempCnt:
                        product[ip[i][1]][0][1] -= tempCnt
                        ans.append(product[ip[i][1]][0][0] * tempCnt)
                        tempCnt = 0
                    elif product[ip[i][1]][0][1] == tempCnt:
                        ans.append(product[ip[i][1]][0][0] * tempCnt)
                        tempCnt = 0
                        del product[ip[i][1]][0]
                    else:
                        ans.append(product[ip[i][1]][0][0] * product[ip[i][1]][0][1])
                        tempCnt -= product[ip[i][1]][0][1]
                        del product[ip[i][1]][0]
        elif ip[i][0] == "return":
            product[ip[i][1]].append([int(ip[i][3]), int(ip[i][2])])
    print(product)
    print(ans)

solution(input)