문제 설명

1. 각 정점에서의 곧바로 연결된 것은 1 촌 1 개를 걸쳐서 연결된것은 2촌 n 개를 걸쳐서 알게된것은 n + 1촌이다.
2. 최단거리를 구하는 문제이다.

풀이

1. 빠르게 답을 알면 종료하기 위해서 우선순위 큐를 사용했다.
2. 답을 구하지 못해서 무한루프를 도는것을 방지하기 위해 visited 배열을 사용했다.
3. x, y 의 점 중 둘 중 x 로 시작해서 x 로 연결된 점은 bfs 방식으로 모두 탐색해서 최단거리를 구했다. (dijkstra)

코드

문제

우리 나라는 가족 혹은 친척들 사이의 관계를 촌수라는 단위로 표현하는 독특한 문화를 가지고 있다. 이러한 촌수는 다음과 같은 방식으로 계산된다. 기본적으로 부모와 자식 사이를 1촌으로 정의하고 이로부터 사람들 간의 촌수를 계산한다. 예를 들면 나와 아버지, 아버지와 할아버지는 각각 1촌으로 나와 할아버지는 2촌이 되고, 아버지 형제들과 할아버지는 1촌, 나와 아버지 형제들과는 3촌이 된다.

여러 사람들에 대한 부모 자식들 간의 관계가 주어졌을 때, 주어진 두 사람의 촌수를 계산하는 프로그램을 작성하시오.

입력

사람들은 1, 2, 3, …, n (1≤n≤100)의 연속된 번호로 각각 표시된다. 입력 파일의 첫째 줄에는 전체 사람의 수 n이 주어지고, 둘째 줄에는 촌수를 계산해야 하는 서로 다른 두 사람의 번호가 주어진다. 그리고 셋째 줄에는 부모 자식들 간의 관계의 개수 m이 주어진다. 넷째 줄부터는 부모 자식간의 관계를 나타내는 두 번호 x,y가 각 줄에 나온다. 이때 앞에 나오는 번호 x는 뒤에 나오는 정수 y의 부모 번호를 나타낸다.

각 사람의 부모는 최대 한 명만 주어진다.

출력

입력에서 요구한 두 사람의 촌수를 나타내는 정수를 출력한다. 어떤 경우에는 두 사람의 친척 관계가 전혀 없어 촌수를 계산할 수 없을 때가 있다. 이 때에는 -1을 출력해야 한다.

예제 입력 1 

9
7 3
7
1 2
1 3
2 7
2 8
2 9
4 5
4 6

예제 출력 1 

3


알고스팟 문제와 동일해서 쉽게 풀었다.


달라진 점이라면 시작점이 0이 아니라는것과 test case 가 여러개라는것


struct node {
int x, y, w;
node (int x, int y, int w): x(x), y(y), w(w) {}
bool operator < (node other) const {
return w > other.w;
}
};

이 부분만 잘 기억해야겠다.


1 -> N 의 최단거리를 구하면 된다.

조건 

1. v1 를 거쳐야 된다.

2. v2 를 거쳐야 된다.

3. 양방향 그래프다.



문제를 보니 민힙을 구현한 다익스트라를 사용해야 될 것 같았다.


풀이 방법


1 -> v1 -> v2 -> N
1 -> v2 -> v1 -> N

2 개중에 낮은 distance 를 출력하면 된다.


둘다 INF 면 -1 를 출력




다익스트라 알고리즘을 정리해보면


u -> v 의 minimum distance 를 구하는 알고리즘.


민힙을 사용해서 구현하면,

현재점을 startVertex, startVertex 에 연결된 점을 endVertex 라고 하자.


u -> startVertex -> endVertex 의 거리가

u -> 이미 간 점 -> endVertex 의 거리보다 작거나 처음 갱신한다면,

민힙에 push 해주고, 거리를 u -> startVertex -> endVertex 의 거리로 갱신해준다.



위의 3줄을 반복하면 정답이 나옴


다익스트라 알고리즘을 2번 돌린다.


while loop에서 모두 처음에

t에 X를 넣어주고, i -> t, t -> i 의 minimum distance를 구한다.

간선이 있으면, 값을 체크해서 넣어주고

visited[t] = true로 만들고 다음 반복으로 간다.


그 다음부터는 더 짧은 거리가 있다면, 업데이트 해준다.


결론적으로 구하는 방법은 i -> t 갈 때, t -> i 갈때 를 구해서 더해준다.


i -> 와 t -> 의 경로는 다를 수 있다.


start -> end 로 갈때의 최단거리를 구하는 문제.


Dijkstra를 사용함.


만약 here -> there을 갈 때, 

int there = edges[here][i].v;

u -> here -> there 가 더빠른게 생기면 업데이트 된다.

if (distance[there] > nextDist) {


1 -> 3 -> 5

4다.


처음 1 -> 5 에 10

3 -> 5 1해가지고 3 + 1 = 4


나무위키에 올라온 사진들을 보며 코드를 작성했다.



MAX값을 2147483647 즉 signed int 의 MAX로 하다보니, 출력에서 꼬여가지고 출력초과과 떴었다.

백준의 출력 용량은 최대 1 MB이다


그 다음에 MAX값을 변경해서 제출하니 시간초과가 뜨는 것이다.

우선순위 큐로 작업하지않고, 리스트에 다 담아서 작업했었음.


그래서 큐에서 값이 변경될 경우에 offer해줬고, 성공했다.




'IT > 알고리즘' 카테고리의 다른 글

BOJ 14852 타일채우기 3 풀이  (0) 2017.10.18
BOJ 2309 일곱난쟁이 풀이  (0) 2017.10.18
BOJ 2065 줄 세우기  (0) 2017.10.18
BOJ 11404 플로이드 Floyd-Warshall  (0) 2017.10.18
BOJ 1717 집합의 표현 disjoint-set  (0) 2017.10.17

+ Recent posts