지형 편집

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

def solution(land, p, q):  # p 추가 비용, q 제거 비용
    land = sorted(each for row in land for each in row)
    cost = (sum(land) - land[0] * len(land)) * q
    answer = cost
    for i in range(1, len(land)):
        if land[i] != land[i - 1]:
            cost += (p * i * (land[i] - land[i - 1]) 
                     - q * (len(land) - i) * (land[i] - land[i - 1]))
            answer = min(answer, cost)
            if answer < cost:
                break
    return answer


print(solution([[1, 2], [2, 3]], 3, 2))
print(solution([[4, 4, 3], [3, 2, 2], [2, 1, 0]], 5, 3))

정확성  테스트
테스트 1 〉	통과 (0.01ms, 10.3MB)
테스트 2 〉	통과 (0.02ms, 10.4MB)
테스트 3 〉	통과 (0.01ms, 10.3MB)
테스트 4 〉	통과 (0.01ms, 10.4MB)
테스트 5 〉	통과 (0.01ms, 10.2MB)
테스트 6 〉	통과 (0.01ms, 10.2MB)
테스트 7 〉	통과 (0.01ms, 10.2MB)
테스트 8 〉	통과 (0.01ms, 10.2MB)
테스트 9 〉	통과 (0.01ms, 10.2MB)
테스트 10 〉	통과 (0.03ms, 10.4MB)
테스트 11 〉	통과 (0.05ms, 10.2MB)
테스트 12 〉	통과 (0.10ms, 10.2MB)
테스트 13 〉	통과 (0.23ms, 10.2MB)
테스트 14 〉	통과 (1.08ms, 10.4MB)
테스트 15 〉	통과 (2.14ms, 10.3MB)
테스트 16 〉	통과 (4.59ms, 10.7MB)
테스트 17 〉	통과 (3.19ms, 10.5MB)
테스트 18 〉	통과 (3.70ms, 10.6MB)
테스트 19 〉	통과 (6.03ms, 10.6MB)
테스트 20 〉	통과 (4.27ms, 10.6MB)
테스트 21 〉	통과 (0.01ms, 10.2MB)
테스트 22 〉	통과 (0.01ms, 10.2MB)
테스트 23 〉	통과 (0.01ms, 10.3MB)
테스트 24 〉	통과 (0.01ms, 10.2MB)
테스트 25 〉	통과 (0.02ms, 10.2MB)
테스트 26 〉	통과 (0.01ms, 10.1MB)
테스트 27 〉	통과 (0.04ms, 10.4MB)
테스트 28 〉	통과 (0.01ms, 10.2MB)
테스트 29 〉	통과 (0.01ms, 10.2MB)
테스트 30 〉	통과 (0.01ms, 10.2MB)
테스트 31 〉	통과 (0.01ms, 10.3MB)
테스트 32 〉	통과 (0.02ms, 10.2MB)
효율성  테스트
테스트 1 〉	통과 (52.92ms, 14.8MB)
테스트 2 〉	통과 (46.32ms, 14.8MB)
테스트 3 〉	통과 (30.79ms, 14.7MB)
테스트 4 〉	통과 (21.64ms, 14.7MB)
테스트 5 〉	통과 (19.58ms, 14.8MB)
테스트 6 〉	통과 (9.11ms, 11.8MB)
테스트 7 〉	통과 (15.71ms, 14.9MB)
테스트 8 〉	통과 (49.49ms, 14.1MB)