5658 보물상자 비밀번호

보물상자 비밀번호

Solution

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

class Solution {
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		int T = Integer.parseInt(br.readLine());
		
		for (int t = 1; t <= T; t++) {
			StringTokenizer st = new StringTokenizer(br.readLine());
			TreeSet<String> set = new TreeSet<>(Collections.reverseOrder());
			int N = Integer.parseInt(st.nextToken());
			int K = Integer.parseInt(st.nextToken());
			char[] c = br.readLine().toCharArray();
			
			for (int i = 0; i < N/4; i++) {
				char last = c[N-1];
				for (int j = N-1; j > 0; j--) {
					c[j] = c[j-1];
				}
				c[0] = last;
//				System.out.println(Arrays.toString(c));
				for (int j = 0; j < N; j+=N/4) {
					String s = "";
					for (int k = j; k < j+N/4; k++) {
						s += c[k];
					}
					set.add(s);
				}
			}
//			System.out.println(set);
			String[] s = set.toArray(new String[set.size()]);
			int answer = Integer.parseInt(s[K-1], 16);
			System.out.println("#" + t + " " + answer);
		}
	}
}

배운점

새롭게 알게된게 많은 문제였다.

이 문제의 첫번째 고민은 생성 가능한 암호를 어디에 담을지
두 번째 고민은 한칸씩 뒤로 밀기를 어떻게 할지
세 번째 고민은 뒤로 밀기를 한 뒤에 어떻게 3개씩 담을지이다.

먼저 어디에 담을지는 다양한 선택지가 있지만 중복은 허용되지 않기 때문에 Set 컬렉션을 사용하는게 좋고 그중 저장하면 정렬이 되는 TreeSet을 사용하고 역순으로 담아 주었다.

그 다음 한칸씩 뒤로 밀어야 되는데 처음엔 앞에서부터 차례대로 담을 생각을 했더니 같은수만 출력이 되어서 당황했다. 뒤에서 부터 담아야 올바르게 담을 수 있고 0번째는 비워두었다가 마지막에 담아준다.

마지막으로 3개씩 담아주기 위해 2중for문을 사용하였다.

여기서 중요한건 TreeSet은 특정 인덱스를 가져오는 메서드가 없기 때문에
Stirng 배열로 변환해야한다.