[프로그래머스-코딩 기초 트레이닝] 전국 대회 선발 고사

Hi, There!
안녕하세요, 바오밥입니다.

 

---

목차

• 문제
• 풀이

 

---

문제

문제 내용

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

프로그래머스

 

---

풀이

나의 풀이

- HashMap, ArrayList 자료구조 활용

import java.util.HashMap;
import java.util.Collections;
import java.util.ArrayList;

class Solution {
    public int solution(int[] rank, boolean[] attendance) {
//         K : rank
//         V : rankIndex
        HashMap<Integer, Integer> map = new HashMap<Integer, Integer>();
        ArrayList<Integer> intList = new ArrayList<Integer>();
        
        for(int i=0; i<rank.length; i++) map.put(rank[i], i);
        
        for(int i=0; i<rank.length; i++)
            if(attendance[i])
                intList.add(rank[i]);

        Collections.sort(intList);
        
        int a = map.getOrDefault(intList.get(0), 0);
        int b = map.getOrDefault(intList.get(1), 0);
        int c = map.getOrDefault(intList.get(2), 0);
               
        int answer = a*10000 + b*100 + c;
        return answer;
    }
}

 

다른 사람의 풀이

- PriorityQueue 자료구조 활용

import java.util.PriorityQueue;

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

    public class Student implements Comparable<Student> {
        int index;
        int number;

        public Student(int index, int number) {
            this.index = index;
            this.number = number;
        }
        
        public int getNumber() {
            return this.number;
        }

        @Override
        public int compareTo(Student obj) {
            if (this.number > obj.getNumber()) return 1;
            else if (this.number < obj.getNumber()) return -1;
            return 0;
        }
    }
}