프로그래머스_직업군 추천하기(JAVA)

https://programmers.co.kr/learn/courses/30/lessons/84325

코딩테스트 연습 - 4주차

문제 설명

개발자가 사용하는 언어와 언어 선호도를 입력하면 그에 맞는 직업군을 추천해주는 알고리즘을 개발하려고 합니다.

아래 표는 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	language	preerence	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"

---

입출력 예 설명

입출력 예 #1

각 직업군 별로 점수를 계산해보면 아래와 같습니다.

아래 사진은 개발자 언어 선호도 나타낸 표입니다.

아래 사진은 개발자가 선호하는 언어의 직업군 언어 점수를 나타낸 표입니다.

따라서 점수 총합이 41로 가장 높은 "HARDWARE"를 return 해야 합니다.

입출력 예 #2

각 직업군 별로 점수를 계산해보면 아래와 같습니다.

아래 사진은 개발자 언어 선호도 나타낸 표입니다.

아래 사진은 개발자가 선호하는 언어의 직업군 언어 점수를 나타낸 표입니다.

점수 총합이 55로 가장 높은 직업군은 "SI" 와 "PORTAL"입니다.
따라서 사전 순으로 먼저 오는  "PORTAL"을 return 해야 합니다.

---

[문제 풀이]

1. table의 각 원소를 공백(" ")을 기준으로 split하여 String 배열 t에 넣어주면 0번째 원소에는 해당 직업군이 들어가게 되고, t[1]에는 5점언어, t[2]에는4점언어, t[3]에는 3점, t[4]에는 2점, t[5]에는 1점 언어가 순서대로 들어가게 된다.

정수변수 sum은 해당 직업군에서 점수 총합을 계산한 값을 넣을 변수이므로 0으로 초기화해준다.

 

2. 위에 1번에서 생성한 배열 t에서 language[j]를 포함하고 있다면, t에서 해당 언어의 인덱스를 찾는다. 배열 t에서 원소의 인덱스를 찾기 위해서는 배열을 리스트로 변경해주고, 해당 리스트에서 인덱스를 검색해야한다. 따라서. Arrays.asList(t)로 리스트 변환을 해주고, .contains(languages[j])로 해당 언어를 포함하고 있는 지 검색한 후, 포함하고 있다면 .indexOf(languages[j])로 인덱스를 찾는다.

찾은 인덱스가 1 이라면 5점 언어인 것이므로 score = 5, 인덱스가 2이면 4점 언어이므로 score=4, ... , 인덱스가 5이면 1점 언어이므로 score=1을 대입해준다. 그 후 점수 총합을 계산해주기 위해 sum += score * preference[j] 수식을 계산한다. 

만약 t가 languages[j]를 포함하지 않는다면 해당 언어의 점수는 0인 것으므로 sum의 크기 변화 없이 다음 언어에 대해서 2 과정을 진행한다.

 

3. 배열 languages의 모든 원소에 대해 2번 과정이 종료되었다면, max와 sum 값을 비교한다. 만약 max 값보다 sum값이 더 크다면 이전에 계산한 직업군 점수 총합보다 현재 직업군 점수의 총합이 더 큰 것이므로 return할 직업군인 answer을 변경해주어야 한다. 따라서 answer에는 현재 계산한 직업군 명이 들어있는 t[0]을 넣어주고 max값을 sum값으로 변경한다.

만약 max와 sum의 크기가 같다면, 사전순으로 비교해 더 앞에 오는 직업군을 넣어준다. 이는 compareTo()를 이용하여 비교한다.

answer.compareTo(t[0])의 값이 0보다 크다면, t[0]이 현재 answer보다 사전순으로 앞에 오는 문자열이므로 answer을 t[0]으로 변경하여주고, 그렇지 않다면 answer이 더 먼저 오는 문자열이므로 변화없이 그대로 둔다.

 

4. 모든 직업군에 대해 위의 과정이 완료된다면, answer에는 점수 총합이 가장 높은 직업군 중 사전순으로 가장 먼저 오는 직업군이 들어있을 것이고, 해당 값을 반환한다.

 

[코드]

import java.util.*;
class Solution {
    public String solution(String[] table, String[] languages, int[] preference) {
        String answer = "";
        int sum = 0, max=0;
        
        for(int i=0;i<table.length;i++){
            String[] t = table[i].split(" ");
            sum=0;
            for(int j=0;j<languages.length;j++){
                if(Arrays.asList(t).contains(languages[j])){
                    int idx = Arrays.asList(t).indexOf(languages[j]);
                    int score=0;
                    if(idx==1)
                        score=5;
                    else if(idx==2)
                        score=4;
                    else if(idx==3)
                        score=3;
                    else if(idx==4)
                        score=2;
                    else if(idx==5)
                        score=1;
                    
                    sum += score*preference[j];
                }
            }
            if(sum>max){
                answer = t[0];
                max=sum;
            }
            else if(sum==max){
                if(answer.compareTo(t[0])>0)
                    answer = t[0];
            }
        }        
        return answer;
    }
}