[python] 등굣길

https://school.programmers.co.kr/learn/courses/30/lessons/42898

---

분명히 다 풀었는데 제출하니까 답이 아니라는거임

진짜 어이없다 왜 안 맞는거야 하고 질문하기를 보니 아주 욕을 많이 먹는 문제였다

puddles로 주는  좌표가 x,y가 아니라 y,x 라고 해서 조건문에서 [i,j] -> [j,i]로 바꾸니까 해결됨

집에서 학교까지 가는 길은 m x n 크기의 격자모양으로 나타낼 수 있습니다. <- 그림 보고 당연히 n행 m열로 만들었는데

가장 왼쪽 위, 즉 집이 있는 곳의 좌표는 (1, 1)로 나타내고 가장 오른쪽 아래, 즉 학교가 있는 곳의 좌표는 (m, n)으로 나타냅니다. 

아니 장난하나....

 

import heapq
 
def solution(n, works):
    answer = 0
    if sum(works) <= n:
        return 0
    max_heap = []
    for work in works:
        heapq.heappush(max_heap, (-work, work))
 
    for _ in range(n):
        work = heapq.heappop(max_heap)[1] - 1
        heapq.heappush(max_heap,(-work,work))
    return sum(list(map(lambda x: x[1]**2,max_heap)))