[hackerrank] Ice Cream Parlor (Python)

문제 (링크)

• 아이스크림 가게에 간 두 친구, 출금한 돈에 딱 맞게 두 개의 아이스크림을 사려고 함
• 쓸 수 있는 돈과 각 아이스크림의 가격이 담긴 리스트가 주어짐
• 가격표의 index는 1부터 시작!
• 가격에 맞게 살 수 있는 아이스크림 가격의 index를 return

파이썬 풀이

def icecreamParlor(m, arr):
    # Write your code here
    pick = []
    for i, v in enumerate(arr[:-1]):
        for j, w in enumerate(arr[i+1:]):
            if v+w == m:
                pick.extend([i+1, i+j+2])
    return pick

 굳이 가격표를 처음부터 순회할 필요가 없다~