[프로그래머스/Lv.0] 전국 대회 선발 고사

1. Input, Output, Example

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

- 전국 대회에 선발된 학생 번호들을 등수가 높은 순서대로 각각 a, b, c번이라고 할 때 10000 × a + 100 × b + c를 반환

 

2. Constraint

1) 3 ≤ rank의 길이 = attendance의 길이 ≤ 100

2) rank[i]는 i번 학생의 선발 고사 등수를 의미한다.

3) rank의 원소는 1부터 n까지의 정수로 모두 서로 다르다.

4) attendance[i]는 i번 학생의 전국 대회 참석 가능 여부를 나타낸다.

5) attendance[i]가 true라면 참석 가능, false면 참석 불가능을 의미한다.

6) attendance의 원소 중 적어도 3개는 true이다.

 

3. Code

1) 첫 코드(2023/04/25)

import java.util.*;
class Solution {
    public int solution(int[] rank, boolean[] attendance) {
        int ans = 0, cnt = 0, r = 1;
        int[] abc = new int[3];
        HashMap<Integer,Integer> m = new HashMap<>();
        for(int i=0 ; i<rank.length ; i++)
            m.put(rank[i],i);
        while(cnt<3){
            if(attendance[m.get(r)])
                abc[cnt++] = m.get(r);
            r++;
        }
        return abc[0]*10000 + abc[1]*100 + abc[2];
    }
}