글 읽기 - 어디가 틀렸는지 모르겠습니다 ㅠㅠ

gsh0910 3년 전 0

만들어본 예제까지도 정답이 나오는데,,

어느 부분이 잘못된건지 모르겠습니다 ㅠㅠ

알고리즘은 1에서 가장 먼 노드를 구하고

그 노드에서 가장 먼 노드를 구한값을 출력하고 있습니다

#include<iostream>
#include<vector>
#include<deque>
#include<algorithm>
using namespace std;

const int max = 1000000 +1;
vector< vector<pair<int,int> > > v;
vector<pair<int,int> > tmp;
deque<pair<int,int> > dq;
int n;

void bfs(int a){
	tmp.clear();
	bool visited[n+1];
	dq.push_back(make_pair(a,0));
	visited[a] = true;
	int l = 0;
	
	while(!dq.empty()){
		int node = dq[l].first;
		int sum = dq[l].second;
		dq.pop_front();
		for(int i=0; i<v[node].size(); i++){
			if(visited[v[node][i].first] == false){
				visited[v[node][i].first] = true;
				dq.push_back(make_pair( v[node][i].first, sum+v[node][i].second) );    
				tmp.push_back(make_pair( v[node][i].first, sum+v[node][i].second) );
			}
		}
		
	}
}
bool compare(pair<int,int> a, pair<int,int> b){
	return a.second > b.second;
}
int main(){
	cin >> n;
	v.resize(n+1);
	
	for(int i=0; i<n; i++){
		int a;
		cin >> a;
		while(1){
			int b , c;
			cin >> b;
			if(b == -1)
				break;
			cin >> c;
			v[a].push_back(make_pair(b,c));
		}
	}

	bfs(1);
	sort(tmp.begin(), tmp.end(), compare);
	bfs(tmp[0].first);
	sort(tmp.begin(),tmp.end(), compare);
	
	cout  << tmp[0].second << '\n';
}