전국 대회 선발 고사

문제 설명

0번부터 n - 1번까지 n명의 학생 중 3명을 선발하는 전국 대회 선발 고사를 보았습니다. 등수가 높은 3명을 선발해야 하지만, 개인 사정으로 전국 대회에 참여하지 못하는 학생들이 있어 참여가 가능한 학생 중 등수가 높은 3명을 선발하기로 했습니다.

각 학생들의 선발 고사 등수를 담은 정수 배열 rank와 전국 대회 참여 가능 여부가 담긴 boolean 배열 attendance가 매개변수로 주어집니다. 전국 대회에 선발된 학생 번호들을 등수가 높은 순서대로 각각 a, b, c번이라고 할 때 10000 × a + 100 × b + c를 return 하는 solution 함수를 작성해 주세요.

제한사항

3 ≤ rank의 길이 = attendance의 길이 ≤ 100
rank[i]는 i번 학생의 선발 고사 등수를 의미합니다.
rank의 원소는 1부터 n까지의 정수로 모두 서로 다릅니다.
attendance[i]는 i번 학생의 전국 대회 참석 가능 여부를 나타냅니다.
attendance[i]가 true라면 참석 가능, false면 참석 불가능을 의미합니다.
attendance의 원소 중 적어도 3개는 true입니다.

나의 코드

import java.util.*;

class Solution {
    public int solution(int[] rank, boolean[] attendance) {
        Map<Integer, Integer> map = new HashMap<>();
        for(int i=0; i<rank.length; i++) {
            if(attendance[i]) {
                map.put(rank[i], i);
            }
        }
        List<Integer> list = new ArrayList<>(map.keySet());
        Collections.sort(list);
        int a = map.get(list.get(0));
        int b = map.get(list.get(1));
        int c = map.get(list.get(2));
        return 10000*a + 100*b + c;
    }
}

느낀 점

미리 정렬을 하고 시작하면 인덱스가 흐트러져서, 다른 방법으로 풀어야하는 문제였다. 일단 원하는 값을 뽑아내기 위해 필요한 요소가 2가지기 때문에 Map을 사용하기로 했고, attendance값이 true일 때만 뽑아 map에 넣어줬다. map의 keyset을 뽑아와서 정렬해준 후, 미리 저장해둔 map이 갖는 인덱스 값을 구해 해결했다. 많이 까다롭고 어려웠던 문제,, 나중에 다시 한번 풀어봐야겠다!