[Baekjoon] 28432_끝말잇기

Silver V

문제(출처: https://www.acmicpc.net/problem/28432)

---

< 끝말잇기 >

 

문제 풀이 

 

? 가 주어지기 전 문자열의 끝 문자와 ? 가 주어진 뒤 문자열의 첫 문자를 저장한 후 문자열 후보 중에서 조건에 맞는 문자열을 찾는다.

 

 my solution (Java)

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.HashSet;
import java.util.Set;

public class _28432_ { // 끝말잇기

	public static void main(String[] args) throws IOException {
		BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));

		int n = Integer.parseInt(bf.readLine());

		Set<String> set = new HashSet<>();
		char before = '0';
		char after = '0';
		boolean flag_before = false;
		boolean flag_after = false;

		for (int i = 0; i < n; i++) {
			String str = bf.readLine();
			if (!str.equals("?")) {
				set.add(str);
				if (flag_before && !flag_after) {
					after = str.charAt(0);
					flag_after = true;
				}
			} else {
				flag_before = true;
			}
			if (!flag_before)
				before = str.charAt(str.length() - 1);

		}

		String result = "";

		int m = Integer.parseInt(bf.readLine());
		if (n == 1) {
			System.out.println(bf.readLine());
		} else {
			for (int i = 0; i < m; i++) {
				String str = bf.readLine();
				if ((str.charAt(0) == before || before == '0')
						&& (str.charAt(str.length() - 1) == after || after == '0') && !set.contains(str)) {
					result = str;
				}
			}
			System.out.println(result);
		}
	}
}

변수)
n : 끝말잇기 기록의 길이
set : 끝말잇기 나온 문자열
before, after : ?가 주어지기 전 문자열의 끝 문자,? 가 주어진 후 문자열의 첫 문자
flag_before, flag_after : ?가 주어지기 전, 후의 체크
str : 문자열
result : 문제의 답
m : 후보 단어의 개수

 

끝말잇기 기록을 입력 받는다. ?가 나오기 전 문자열의 끝 문자와 ? 가 나온 후의 문자열의 첫 문자를 각각 저장한다. 후보를 입력받으면서 문자열의 첫 문자와 끝 문자가 앞에서 구한 문자와 일치한다면 그 값을 출력한다. 입력받을 때? 가 맨 앞과 맨 뒤에 입력될 수 있으므로 후보 문자열에서 답을 구할 때 조건을 추가한다.

---

---