프로그래머스 - 전국 대회 선발 고사 (JAVA)

---

고민에 고민을 하다가 

TreeMap을 사용하면 상당히 간단하게 풀 수 있다는걸 알고 활용해서 풀어봤다

처음엔 HashMap을 썼었는데 정렬처리가 안되기 때문에 테스트케이스에서 틀리는 경우가 있었다

그래서 TreeMap을 활용해서 풀게 되었다.

 

import java.util.*;

class Solution {
    public int solution(int[] rank, boolean[] attendance) {
        
         TreeMap<Integer, Integer> treeMap = new TreeMap<>();

        for(int i = 0; i < rank.length; i++){
            if(attendance[i]){
                treeMap.put(rank[i] , i);
            }
        }
        List<Integer> list = new ArrayList<>(treeMap.values());

        return 10000*list.get(0) + 100*list.get(1) +  list.get(2);


    }
}