코테 Lv0 - 전국 대회 선발 고사

문제 설명

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입니다.

입출력 예

rank	attendance	result
[3, 7, 2, 5, 4, 6, 1]	[false, true, true, true, true, false, false]	20403
[1, 2, 3]	[true, true, true]	102
[6, 1, 5, 2, 3, 4]	[true, false, true, false, false, true]	50200

 

입출력 예 #1

예제 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

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

 

입출력 예 #3

예제 3번에서 1, 2, 3등은 각각 1번, 3번, 4번 학생이지만 세 학생 모두 전국 대회에 참석 불가능합니다.

다음으로 4, 5, 6등은 각각 5번, 2번, 0번 학생이고 세 학생 모두 전국 대회에 참석 가능하므로 각각 1순위, 2순위, 3순위로 선발됩니다.

따라서 10000 × 5 + 100 × 2 + 0 = 50200을 return 합니다.

제출코드

import java.util.Comparator;
import java.util.stream.IntStream;

class Solution {
    public int solution(int[] rank, boolean[] attendance) {
        int[] sortedRank = IntStream.range(0, rank.length)
            .filter(i -> attendance[i])
            .boxed()
            .sorted(Comparator.comparingInt(i -> rank[i]))
            .limit(3)
            .mapToInt(i -> i)
            .toArray();
        return (10000 * sortedRank[0]) + (100 * sortedRank[1]) + sortedRank[2];
    }
}

1. rank의 length만큼에 index를 가진 IntStream을 생성하고 attendance의 값이 true(대회 참가)인 index만 필터링합니다.
2. boxed() 메서드를 사용해 Integer 형태로 변환 후 rank의 값으로 오름차순 정렬합니다.
3. 최대 3개만 배열로 반환하고 주어신 식을 계산하여 반환합니다.

다른 풀이

import java.util.PriorityQueue;

class Solution {
    public int solution(int[] rank, boolean[] attendance) {
        PriorityQueue<Integer> que = new PriorityQueue<>((a, b) -> rank[a] - rank[b]);
        for (int i = 0; i < attendance.length; i++) {
            if (attendance[i])
                que.add(i);
        }
        return que.poll() * 10000 + que.poll() * 100 + que.poll();
    }
}

import java.util.ArrayList;
import java.util.TreeMap;

class Solution {
    public int solution(int[] rank, boolean[] attendance) {
        TreeMap<Integer, Integer> tree = new TreeMap<>();
        for (int i = 0; i < rank.length; i++) {
            if (attendance[i] == true) tree.put(rank[i], i);
        }

        ArrayList<Integer> list = new ArrayList<>();
        for (Integer key : tree.keySet()) {
            if (list.size() == 3) break;
            list.add(tree.get(key)); 
        }
        return list.get(0) * 10000 + list.get(1) * 100 + list.get(2);
    }
}

 

PriorityQueue(우선순위 큐)와 TreeMap를 이용해서 순서를 지정한 것이 인상적이었습니다.

역시 자료구조를 많이 잘 사용할수록 간단해지는 것 같습니다.

---

출처 : https://school.programmers.co.kr/learn/courses/30/lessons/181851

프로그래머스