[Java] 백준 31423 - 신촌 통폐합 계획

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

---

풀이

 

읽어보면 연결 리스트가 떠오르는 문제

다만 평범한 연결 리스트로 풀면 뒤에 이어 붙이는 과정에서 매번 head에서

부터 tail까지 탐색해 나가기 때문에 총 시간 복잡도가 O(N^2)이 되어버림

그래서 노드마다 tail 값을 두고 매번 next와 tail 값을 잘 업데이트해 줘야 함  

 

다른 사람들 풀이 보니 그래프로 생각해서 DFS로도 풀이가 가능한 것 같다

 

 

코드

import java.io.*;
import java.util.*;

public class Main {
	public static void main(String[] args) throws IOException {

		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		int N = Integer.parseInt(br.readLine());
		String[] s = new String[N + 1];
		int[] next = new int[N + 1];
		int[] tail = new int[N + 1];

		for (int i = 1; i <= N; i++) {
			s[i] = br.readLine();
			next[i] = i;
			tail[i] = i;
		}

		int cur = -1;

		for (int i = 0; i < N - 1; i++) {
			StringTokenizer st = new StringTokenizer(br.readLine());
			int left = Integer.parseInt(st.nextToken());
			int right = Integer.parseInt(st.nextToken());

			next[tail[left]] = right;
			tail[left] = tail[right];
			cur = left;
		}

		StringBuilder sb = new StringBuilder();

		for (int i = 0; i < N; i++) {
			sb.append(s[cur]);
			cur = next[cur];
		}

		System.out.println(sb);
		br.close();
	}
}

 

테스트 케이스

Input:
5
a
b
c
d
e
4 5
3 4
2 3
1 2

Expected output:
abcde


Input:
6
a
b
c
d
e
f
5 6
4 5
2 3
1 2
1 4

Expected output:
abcdef


Input:
2
a
b
2 1

Expected output:
ba