알고리즘 문제풀이/Python
[백준] 1516 게임 개발
언호
2022. 7. 4. 00:56
📖 문제
🧑🏻💻 풀이 과정
1) 문제 접근 및 이해
건물을 짓기 위해 필요한 건물의 개수와 해당 건물이 완성되면 다음에 건설 가능한 건물의 목록을 리스트로 관리하였습니다.
건물을 전체적으로 탐색하여 바로 건설이 가능한 목록을 찾은 후 해당 건물이 완성되면 건설이 가능한 다음 건물을 찾았습니다.
그 건물을 짓는데 필요한 앞 건물의 개수 카운트를 1씩 감소시켰고, 0이 되었을 때 건물을 짓도록 하였습니다.
2) 알고리즘
- 위상정렬
3) 풀이 코드
사용 언어 - Python
import sys
from collections import deque
sys.stdin = open('input.txt')
N = int(sys.stdin.readline()) # 건물 수
linked = [[] for _ in range(N)] # 각 건물이 완성되면 지을 수 있는 건물 목록
need = [0] * N # 건물 짓는데 필요한 앞 건물 개수
times = [0] * N # 건물 짓는데 소요 시간
answers = [0] * N # 건물 완성 시간
for idx in range(N):
infos = list(map(int, sys.stdin.readline().split()))
times[idx] = infos[0] # 해당 건물 짓는데 소요 시간
for i in range(1, len(infos)-1):
linked[infos[i]-1].append(idx) # 해당 건물을 짓기 위해 필요한 건물 기록
need[idx] += 1 # 건물 짓는데 필요한 앞 건물 개수 추가
q = deque()
for idx in range(N): # 앞 건물 없이 바로 지을 수 있으면
if not need[idx]: # 큐에 추가
q.append(idx)
answers[idx] = times[idx]
while q:
node = q.popleft()
for next in linked[node]:
# 다음 건물 짓는데 필요한 건물 수 감소
need[next] -= 1
# 다음 건물 총 소요시간은 현재 건물 시간 + 다음 건물 시간
answers[next] = max(answers[next], answers[node] + times[next])
# 다음 건물을 짓기 위해 필요한 건물을 모두 지었다면, 큐에 추가
if not need[next]:
q.append(next)
for i in range(N):
print(answers[i])
🔗 문제 링크
- https://www.acmicpc.net/problem/1516
1516번: 게임 개발
첫째 줄에 건물의 종류 수 N(1 ≤ N ≤ 500)이 주어진다. 다음 N개의 줄에는 각 건물을 짓는데 걸리는 시간과 그 건물을 짓기 위해 먼저 지어져야 하는 건물들의 번호가 주어진다. 건물의 번호는 1부
www.acmicpc.net
※ 오류 및 오타, 다른 의견이 있는 경우 댓글을 남겨주시면 감사하겠습니다