글 읽기 - 어디가 문제일까요?

illiboy1212 4년 전 0

예제는 맞게 나오고, w를 시작점으로 해서 최장거리를 구하는 방식으로 접근하여 보았습니다

#include <cstdio>
#include <iostream>
#include <utility>
#include <algorithm>
#include <queue>
#include <set>
#include <memory.h>


#define MAX_N 1000 + 10

using namespace std;

int arr[MAX_N];
int d[MAX_N][MAX_N];
int visited[MAX_N];

priority_queue< pair<int, int > > pq;

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

	int z;
	cin >> z;

	int n, k;
	int w;

	while(z--)
	{
		memset(visited, 0, sizeof(visited));

		cin >> n >> k;
		for (int i = 1; i <=n; i++)
		{
			cin >> arr[i];
		}

		for (int i = 0; i < k; i++)
		{
			int a, b;
			cin >> a >> b;
			d[a][b] = 1;
		}

		cin >> w;

		pq.push(make_pair(arr[w] , w ));

		int sum = 0;

		bool update = 0;


		while(1)
		{
			pair<int, int> tmp = pq.top();
			pq.pop();

			sum += tmp.first;

			//cout << tmp.second << " " << tmp.first<< endl;

			update = 0;

			for (int i = 1; i <= n; i++)
			{
				if (d[i][tmp.second] && visited[i]==0)
				{
					update = 1;
					pq.push(make_pair(arr[i], i));
					visited[i]=1;
				}
			}

			if (update==0)
			{
				break;
			}
		}

		cout << sum << endl;
	}
}

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

어디가 문제일까요?

illiboy1212 4년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말