개발자가 사용하는 언어와 언어 선호도를 입력하면 그에 맞는 직업군을 추천해주는 알고리즘을 개발하려고 합니다.
아래 표는 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
1 ≤ languages의 길이 ≤ 9
preference의 길이 = languages의 길이
preference의 i번째 원소는 languages의 i번째 원소의 언어 선호도입니다.
return 할 문자열은 "SI", "CONTENTS", "HARDWARE", "PORTAL", "GAME" 중 하나입니다.
table | languages | perperence | 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" |
import java.util.*;
class Solution {
public String solution(String[] table, String[] languages, int[] preference) {
String answer = "";
HashMap<String, HashMap<String, Integer>> company = new HashMap<>();
int sum = 0;
int max = 0;
for(int i = 0; i < table.length; i++) {
int num = 5;
HashMap<String, Integer> lang = new HashMap<>();
for(int j = 1; j < 6; j++) {
lang.put(table[i].split(" ")[j], num--);
}
company.put(table[i].split(" ")[0], lang);
}
for(String com : company.keySet()) { // 기업 분야
sum = 0; //초기화
for(String language : company.get(com).keySet()) { // 언어들
for(int i = 0; i < languages.length; i++) {
if(language.equals(languages[i])) { //언어가 같은면
sum += preference[i] * company.get(com).get(language); //점수
}
}
}
if(sum > max) {
max = sum;
answer = com;
}else if(sum == max){
if(answer.compareTo(com) > 0) { // answer 가 com보다 사전적 의미가 앞서면
answer = com;
}
}
}
return answer;
}
}
HashMap<String, HashMap<String, Integer>>
의 첫 번째 hashmap에는 SI, CONTENTS .. 의 기업분야? 를 넣어주고, 두 번째 hashmap은 언어들을 넣어 준다.company.keySet()
은 company의 key값만 가지고 온다. 즉, SI, CONTENTS, HARDWARE ... 이 com안에 담긴다.company.get(com).keySet()
여기서 get은 인자(com)로 전달된 key에 해당하는 value를 리턴해주고 그 value의 key값 즉, com = SI면 JAVA, JAVASCRIPT ...등이 language에 담긴다.company.get(com).get(language)
의 코드는 인자로 전달된 com의 key(SI, CONTENTS ...)에 해당하는 value(JAVA,JAVASCRIPT...)의 value(5점,4점,3점,2점,1점)값을 가져온다.answer.compareTo(com)
answer == com return 0;, answer < com return -1;, answer > com return 1; 사전적 의미를 구하기 위해 사용함.