Notice
Recent Posts
Recent Comments
Link
일 | 월 | 화 | 수 | 목 | 금 | 토 |
---|---|---|---|---|---|---|
1 | 2 | 3 | 4 | 5 | 6 | 7 |
8 | 9 | 10 | 11 | 12 | 13 | 14 |
15 | 16 | 17 | 18 | 19 | 20 | 21 |
22 | 23 | 24 | 25 | 26 | 27 | 28 |
29 | 30 |
Tags
- 스택
- 백준
- level3
- dict
- 그리디
- 코딩테스트
- 가상메모리
- BFS
- level0
- N과M
- 다이나믹 프로그래밍
- DP
- 수학
- 프로그래머스
- programmers
- 딕셔너리
- 운영체제
- 재귀
- MYSQL
- 브루트포스
- python
- 가상메모리 관리
- level2
- dfs
- BOJ
- 다익스트라
- 파이썬
- level1
- 구현
- 힙
Archives
- Today
- Total
동캄의 코딩도장
프로그래머스 level3 [야근 지수] 파이썬 본문
https://programmers.co.kr/learn/courses/30/lessons/12927?language=python3
# 프로그래머스 야근 지수
import heapq
def solution(n, works):
answer = 0
q = []
s = sum(works)
for work in works:
heapq.heappush(q, -work)
if n >= s:
answer = 0
else:
for i in range(n):
work = heapq.heappop(q)
work += 1
heapq.heappush(q, work)
for work in q:
answer += (work**2)
return answer
max힙을 이용하여 간단히 구현하였다.
'코테 > 프로그래머스' 카테고리의 다른 글
프로그래머스 level3 [최고의 집합] 파이썬 (0) | 2022.04.11 |
---|---|
프로그래머스 level3 [멀리 뛰기] 파이썬 (0) | 2022.04.11 |
프로그래머스 level3 [순위] 파이썬 (0) | 2022.04.09 |
프로그래머스 level3 [ 2 x n 타일링] 파이썬 (0) | 2022.04.08 |
프로그래머스 level3 [가장 먼 노드] 파이썬 (0) | 2022.04.08 |