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

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

프로그래머스

 

 

 

 

 

 

 

정답 

 

import java.util.*;

class Solution {
    public int solution(int[] rank, boolean[] attendance) {
        
        //key-등수 value-인덱스
        HashMap<Integer,Integer> map = new HashMap<>();
        
        for(int i=0; i<attendance.length; i++)
        {
            if(attendance[i])
            {
                map.put(rank[i],i);
            }
        }
        
        List<Integer> keySet = new ArrayList<>(map.keySet());
        
        //등수(key)로 오름차순 정렬
        Collections.sort(keySet);
        
        //key값 배열에 저장
        int temp[] = new int[map.size()];
        int idx=0;
        
        for(int k: keySet)
        {
            temp[idx++] = k;
        }
        
        int a = map.get(temp[0]);
        int b = map.get(temp[1]);
        int c = map.get(temp[2]);
        
        int answer = 10000*a + 100*b + c;
        
        return answer;
    }
}

 

 

다른 사람의 풀이 

 

1)

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);
    }
}

 

 

TreeMap

 

- 이진트리를 기반으로 Map 인터페이스를 구현한 컬렉션 클래스 

- HashMap과 마찬가지로 키와 값(Key-Value)쌍을 저장한다. 

- HashMap은 키의 해시값을 기반으로 해시테이블을 구축하는 반면 TreeMap은 키 값을 이용해서 이진 트리를 구축하고, 이진 트리의 노드에 값을 엔트리(Entry) 형태로 저장한다. 

- 이진트리를 구축했기 때문에 키 값에 따라 정렬된 상태로 저장한다.