[Baekjoon] 1283_단축키 지정

Silver I

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

---

< 단축키 지정 >

 

문제 풀이 

 

먼저 단어를 입력받으면서 단어의 첫 글자가 단축키인지 확인하여 단축키가 아니라면 단축키로 설정한다. 모든 단어의 첫 글자가 이미 단축키라면 왼쪽부터 차례대로 알파벳을 보면서 단축키로 지정 안 된 것이 있다면 단축키로 설정한다.

 

 

 my solution (Java)

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.Set;
import java.util.StringTokenizer;

public class _1283_ { // 단축키 지정

	public static void main(String[] args) throws IOException {
		BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
		BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
		StringTokenizer st;

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

		Set<Character> set = new HashSet<>();
		ArrayList<String> list = new ArrayList<>();
		boolean flag = false;

		for (int i = 0; i < n; i++) {
			st = new StringTokenizer(bf.readLine());
			flag = false;
			while (st.hasMoreTokens()) {
				String str = st.nextToken();
				if (flag) {
					bw.write(str+" ");
				} else if (!set.contains(Character.toUpperCase(str.charAt(0)))) {
					set.add(Character.toUpperCase(str.charAt(0)));
					str = "[" + str.charAt(0) + "]" + str.substring(1, str.length());
					if (!list.isEmpty()) {
						for (int j = 0; j < list.size(); j++) {
							bw.write(list.get(j) + " ");
						}
						list.clear();
					}
					bw.write(str + " ");
					flag = true;
				} else {
					list.add(str);
				}
			}
			if (!flag) {
				for (int j = 0; j < list.size(); j++) {
					String str = list.get(j);
					if (!flag) {
						for (int k = 0; k < str.length(); k++) {
							if (!set.contains(Character.toUpperCase(str.charAt(k)))) {
								set.add(Character.toUpperCase(str.charAt(k)));
								str = str.substring(0, k) + "[" + str.charAt(k) + "]"
										+ str.substring(k + 1, str.length());
								flag = true;
								break;
							}
						}
					}
					bw.write(str + " ");
				}
			}
			list.clear();
			bw.write("\n");
		}
		bw.flush();
	}
}

변수)
n : 옵션의 개수
set : 단축키 모음
list : 하나의 옵션 문자열 저장
flag : 단축키 설정 여부
str : 문자열

 

하나의 옵션에 대해 단어를 입력받아 단어의 첫 글자가 단축키인지 아닌지 확인한다. 단축키가 아니라면 단축키로 설정한다. 단축키로 설정했으면 나머지는 확인할 필요 없으므로 그냥 문자열을 출력한다.

 

만약 모든 단어의 첫 글자가 단축키로 설정되어 있다면 단어를 왼쪽부터 살피면서 단축키를 확인해야 한다. 단어를 왼쪽부터 차례대로 보면서 단축키가 아니라면 단축키로 설정하고 위와 같이 나머지는 확인할 필요 없으므로 문자열을 출력한다.

---

문자열을 출력하는데 원래 문자열과 일치하도록 공백을 잘 출력해줘야 한다.

---