[programmers] 전국 대회 선발 고사 - Java

문제 설명

제한사항

• 3 ≤ rank의 길이 = attendance의 길이 ≤ 100

• rank[i]는 i번 학생의 선발 고사 등수를 의미합니다.

• rank의 원소는 1부터 n까지의 정수로 모두 서로 다릅니다.

• attendance[i]는 i번 학생의 전국 대회 참석 가능 여부를 나타냅니다.
• attendance[i]가 true라면 참석 가능, false면 참석 불가능을 의미합니다.

• attendance의 원소 중 적어도 3개는 true입니다.

입출력 예

입출력 예 설명

• 예제 1번에서 1등은 6번 학생이지만 전국 대회에 참석 불가능하므로 넘어갑니다. 2등은 2번 학생이고 전국 대회에 참석 가능하므로 1순위로 선발됩니다. 3등은 0번 학생이고 전국 대회에 참석 불가능 하므로 넘어갑니다. 4등과 5등은 각각 4번, 3번 학생이고 두 학생 모두 전국 대회에 참석 가능하므로 각각 2순위, 3순위로 선발됩니다. 3명을 선발하였으므로 6등과 7등은 확인하지 않아도 됩니다. 따라서 10000 × 2 + 100 × 4 + 3 = 20403을 return 합니다.

• 예제 2번에서 1, 2, 3등은 각각 0번, 1번, 2번 학생이고 세 학생 모두 전국 대회에 참석 가능하므로 각각 1순위, 2순위, 3순위로 선발됩니다. 따라서 10000 × 0 + 100 × 1 + 2 = 102를 return 합니다.

• 예제 3번에서 1, 2, 3등은 각각 1번, 3번, 4번 학생이지만 세 학생 모두 전국 대회에 참석 불가능합니다. 다음으로 4, 5, 6등은 각각 5번, 2번, 0번 학생이고 세 학생 모두 전국 대회에 참석 가능하므로 각각 1순위, 2순위, 3순위로 선발됩니다. 따라서 10000 × 5 + 100 × 2 + 0 = 50200을 return 합니다.

solution.java

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> number = new ArrayList<>(map.keySet());
        Collections.sort(number);
        
        return 10000*map.get(number.get(0)) + 100*map.get(number.get(1)) + map.get(number.get(2));
    }
}

핵심 키워드

• Map은 선언 시 <key, value>로 값을 넣는다. 이를 이용해 attendance가 true인 등수를 key로, 인덱스를 value로 put한다.

• Java에서 HashMap을 정렬하려면, key 또는 value 값을 기준으로 정렬을 할 수 있다.
• key값을 기준으로 정렬하려면 map 의 keySet을 이용하여 정렬한다.
• 오름차순 시에는 Collection.sort(), 내림차순 시에는 Collection.reverse() 메소드를 사용하여 정렬한다.

List<Integer> keySet = new ArrayList<>(map.keySet());
Collections.sort(keySet);

• Value 값을 기준으로 정렬할 때는 comparator를 사용하여 정렬한다.
• comparator는 람다 표현식으로 간단하게 표현할 수도 있다.

List<Integer> keySet = new ArrayList<>(map.keySet());
keySet.sort((o1, o2) ->map.get(o1).compareTo(map.get(o2));

결론!