[백준 1516번] 게임 개발

https://www.acmicpc.net/problem/1516 

1516번: 게임 개발

풀이) 

위상정렬을 이용해서 해결할 수 있었다. 지어야 하는 직전 건물 중 가장 늦게 끝나는 건물을 기준으로 minCost를 구해주면 된다.

#include <iostream>
#include <vector>
#include <queue>
using namespace std;

int N, tmp;
int cost[501], inDegree[501], minCost[501];
vector <int> v[501];
queue <int> q;

void topologySort()
{
	for (int i = 1; i <= N; i++)
		if (inDegree[i] == 0)
		{
			q.push(i);
			minCost[i] = cost[i];
		}

	while (!q.empty())
	{
		int now = q.front();
		q.pop();
		for (int i = 0; i < v[now].size(); i++)
		{
			int next = v[now][i];
			minCost[next] = max(minCost[next], minCost[now] + cost[next]);
			if (--inDegree[next] == 0)
				q.push(next);
		}
	}
}

int main()
{
	ios::sync_with_stdio(0);
	cin.tie(0); cout.tie(0);

	cin >> N;
	for (int i = 1; i <= N; i++)
	{
		cin >> cost[i];
		while (cin >> tmp && tmp != -1)
		{
			v[tmp].push_back(i);
			inDegree[i]++;
		}
	}
	topologySort();

	for (int i = 1; i <= N; i++)
		cout << minCost[i] << "\n";
	return (0);
}

 

 

제 코드는 절대 완벽하지 않습니다

더 좋은 풀이방법 혹은 코드에 관한 질문이 있으시면 언제든 댓글주세요!