프로그래머스 위클리 챌린지 4주차를 풀었다 !
이거 분명 상반기 어떤 코테에서 풀었던 문젠데 ! 3주차에 갑자기 난이도 올리더니,, 다시 쉬워졌다,, 코테에서 내 기억에 아마 1번이지 않았나 싶은데,, 그래도 다시 한 번 풀어보았다 !
문제 설명
개발자가 사용하는 언어와 언어 선호도를 입력하면 그에 맞는 직업군을 추천해주는 알고리즘을 개발하려고 합니다.
아래 표는 5개 직업군 별로 많이 사용하는 5개 언어에 직업군 언어 점수를 부여한 표입니다.
| 점수 | SI | CONTENTS | HARDWARE | PORTAL | GAME |
|---|---|---|---|---|---|
| 5 | JAVA | JAVASCRIPT | C | JAVA | C++ |
| 4 | JAVASCRIPT | JAVA | C++ | JAVASCRIPT | C# |
| 3 | SQL | PYTHON | PYTHON | PYTHON | JAVASCRIPT |
| 2 | PYTHON | SQL | JAVA | KOTLIN | C |
| 1 | C# | C++ | JAVASCRIPT | PHP | JAVA |
예를 들면, SQL의 SI 직업군 언어 점수는 3점이지만 CONTENTS 직업군 언어 점수는 2점입니다. SQL의 HARDWARE, PORTAL, GAME 직업군 언어 점수는 0점입니다.
직업군 언어 점수를 정리한 문자열 배열 table, 개발자가 사용하는 언어를 담은 문자열 배열 languages, 언어 선호도를 담은 정수 배열 preference가 매개변수로 주어집니다. 개발자가 사용하는 언어의 언어 선호도 x 직업군 언어 점수의 총합이 가장 높은 직업군을 return 하도록 solution 함수를 완성해주세요. 총합이 같은 직업군이 여러 개일 경우, 이름이 사전 순으로 가장 빠른 직업군을 return 해주세요.
제한사항
- table의 길이 = 5
- table의 원소는 "직업군 5점언어 4점언어 3점언어 2점언어 1점언어"형식의 문자열입니다. 직업군, 5점언어, 4언어, 3점언어, 2점언어, 1점언어는 하나의 공백으로 구분되어 있습니다.
- table은 모든 테스트케이스에서 동일합니다.
- 1 ≤ languages의 길이 ≤ 9
- languages의 원소는 "JAVA", "JAVASCRIPT", "C", "C++" ,"C#" , "SQL", "PYTHON", "KOTLIN", "PHP" 중 한 개 이상으로 이루어져 있습니다.
- languages의 원소는 중복되지 않습니다.
- preference의 길이 = languages의 길이
- 1 ≤ preference의 원소 ≤ 10
- preference의 i번째 원소는 languages의 i번째 원소의 언어 선호도입니다.
- return 할 문자열은 "SI", "CONTENTS", "HARDWARE", "PORTAL", "GAME" 중 하나입니다.
입출력 예
| table | languages | preference | result |
|---|---|---|---|
| ["SI JAVA JAVASCRIPT SQL PYTHON C#", "CONTENTS JAVASCRIPT JAVA PYTHON SQL C++", "HARDWARE C C++ PYTHON JAVA JAVASCRIPT", "PORTAL JAVA JAVASCRIPT PYTHON KOTLIN PHP", "GAME C++ C# JAVASCRIPT C JAVA"] | ["PYTHON", "C++", "SQL"] | [7, 5, 5] | "HARDWARE" |
| ["SI JAVA JAVASCRIPT SQL PYTHON C#", "CONTENTS JAVASCRIPT JAVA PYTHON SQL C++", "HARDWARE C C++ PYTHON JAVA JAVASCRIPT", "PORTAL JAVA JAVASCRIPT PYTHON KOTLIN PHP", "GAME C++ C# JAVASCRIPT C JAVA"] | ["JAVA", "JAVASCRIPT"] | [7, 5] | "PORTAL" |
풀이 방법
먼저 string 배열로 입력받는 table을 활용하기 좋은 table 형태로 구성해주어야 한다. 나는 Job이라는 클래스를 만들고 내부에 직군 이름과 언어를 점수 순서대로 리스트로 넣어주었다.
또한 언어를 넘기면 해당 직군에서의 점수를 얻을 수 있고, 이 점수와 개인의 언어리스트와 선호도를 넘기면 개인의 해당 직군에 대한 점수가 속성에 셋팅된다.
이 점수들을 기준으로 최종 결과가 정해지기 때문에 이 점수를 기준으로 정렬해준다. 만일 점수가 같다면 직군 이름의 사전순으로 정렬될 수 있도록 기준을 Comparator의 compare에 오버라이드 해주었다.
마지막으로 맨 앞에 정렬된 직군의 이름을 반환해주었다.
코드
import java.util.*;
class Solution {
public static class Job{
int size = 5;
String name;
ArrayList<String> languages;
int score;
Job(String table){
String[] temp = table.split(" ");
this.name = temp[0];
this.languages = new ArrayList<>();
this.languages.addAll(Arrays.asList(temp).subList(1, temp.length));
}
int getLanguageScore(String language){
int index = languages.indexOf(language);
if(index == -1) return 0;
else return size - languages.indexOf(language);
}
void setScore(String[] languages, int[] preferences){
int score = 0;
for(int i=0; i<languages.length; i++){
score += (getLanguageScore(languages[i]) * preferences[i]);
}
this.score = score;
}
}
public String solution(String[] table, String[] languages, int[] preference) {
ArrayList<Job> jobs = new ArrayList<>();
Table temp;
for(String str : table){
temp = new Job(str);
temp.setScore(languages, preference);
jobs.add(temp);
}
jobs.sort(new Comparator<Table>() {
@Override
public int compare(Job o1, Job o2) {
if(o1.score == o2.score) return o1.name.compareTo(o2.name);
return o2.score - o1.score;
}
});
return jobs.get(0).name;
}
}