[프로그래머스] 전국 대회 선발고사 (이차원 배열 정렬) - JAVA

프로그래머스 _ 코딩 기초 트레이닝 DAY 21 - (2)

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

프로그래머스

문제 | 0번부터 n - 1번까지 n명의 학생 중 3명을 선발하는 전국 대회 선발 고사를 보았습니다. 등수가 높은 3명을 선발해야 하지만, 개인 사정으로 전국 대회에 참여하지 못하는 학생들이 있어 참여가 가능한 학생 중 등수가 높은 3명을 선발하기로 했습니다. 각 학생들의 선발 고사 등수를 담은 정수 배열 rank와 전국 대회 참여 가능 여부가 담긴 boolean 배열 attendance가 매개변수로 주어집니다. 전국 대회에 선발된 학생 번호들을 등수가 높은 순서대로 각각 a, b, c번이라고 할 때 10000 × a + 100 × b + c를 return 하는 solution 함수를 작성해 주세요.

 

입력 #1 | rank = [3, 7, 2, 5, 4, 6, 1], attendance = [false, true, true, true, true, false, false]

출력 #1 | 20403

---

>> 풀이 << 

이차원 배열을 선언한 후, rank와 그의 index를 저장해주었다. 그렇게 하면 sort를 해도 인덱스를 보존할 수 있음!

문제는 이차원 배열을 어떻게 오름차순 정렬을 할 수 있냐였다.

 

• 이차원배열 정렬하기 (첫번째 열을 기준으로) -Arrays.sort(ans, (a, b) -> {return a[0] - b[0];});
  • Arrays.sort(ans, (a, b) -> {return a[0] - b[0];});처럼 람다식으로 쓸 수 있다. 해당 코드는 첫번째 열을 기준으로 정렬하는 명령어이다.

---

코드는 아래와 같다.

import java.util.*;
class Solution {
    public int solution(int[] rank, boolean[] attendance) {
        int size = 0;
        for (int i = 0 ; i < rank.length ; i++) {
            if (attendance[i]) size++;
        }
        
        int[][] ans = new int[size][2];
        int j = 0;
        for (int i = 0 ; i < rank.length ; i++) {
            if (attendance[i]) {
                ans[j][0] = rank[i];
                ans[j++][1] = i;
            }
        }                                                                                         
        Arrays.sort(ans, (a, b) -> {return a[0] - b[0];});
        
        return 10000 * ans[0][1] + 100 * ans[1][1] + ans[2][1];
    }
}