[SWEA] 보물상자 비밀번호

https://swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AWXRUN9KfZ8DFAUo

SW Expert Academy

🏗️ 설계

• String을 입력받은 후 N / 4의 크기만큼 4개로 잘라 4개의 수를 Integer로 변환하여 Set에 넣음으로써 중복을 제거합니다.
• String의 맨 앞글자를 뒤로 옮기는 작업을 하며 위와 같은 행동을 반복 합니다.
• 총 3번 진행하면 초기 상태와 똑같아 지므로 이 때 Set에 있는 값을 ArrayList에 넣은 후 내림차순으로 Sort 하고, K번째 위치에 있는 값을 출력합니다.

📋 코드

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.Collections;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import java.util.StringTokenizer;

public class Solution_5658_보물상자_비밀번호 {
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
		StringBuilder sb = new StringBuilder();
		int testCase = Integer.parseInt(br.readLine());

		for (int tc = 1; tc <= testCase; tc++) {
			sb.append("#").append(tc).append(" ");
			Set<Integer> set = new HashSet<>();
			StringTokenizer st = new StringTokenizer(br.readLine());
			int size = Integer.parseInt(st.nextToken());
			int target = Integer.parseInt(st.nextToken());

			String str = br.readLine();

			for (int swap = 0; swap < size / 4; swap++) {
				for (int i = 0; i < size; i += size / 4) {
					set.add(Integer.parseInt(str.substring(i, i + size / 4), 16));
				}
				str = str.concat(Character.toString(str.charAt(0))).substring(1, 1 + size);
			}
			
			List<Integer> list = new ArrayList<>(set);
			
			Collections.sort(list, (o1, o2) -> {
				return o2 - o1;
			});
			
			sb.append(list.get(target - 1)).append("\n");
		}
		bw.write(sb.toString());
		bw.flush();
		bw.close();
		br.close();
	}
}

⏰ 사용한 메모리 / 시간

• Memory : 27064KB
• Time : 177ms