Weekly Challenge 4주차

문제 설명

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

아래 표는 5개 직업군 별로 많이 사용하는 5개 언어에 직업군 언어 점수를 부여한 표입니다.

 

 

예를 들면, 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" 중 하나입니다.

 

입출력 예

 

 

import java.util.Arrays;
import java.util.Collections;

class Solution {
    public String solution(String[] table, String[] languages, int[] preference) {
        
        Arrays.sort(table, Collections.reverseOrder());
        
        int max = 0;
		int answer = 0;
		
		String[][] table2 = new String[5][];
		
		for(int i=0; i<table.length; i++) {
			table2[i] = table[i].split(" ");
		}
		
		for(int i=0; i<table2.length; i++) {
			int score = 0;
			int total = 0;
			for(int k=0; k<languages.length; k++) {
				
				for(int j=1; j<table2[i].length; j++) {
					if(table2[i][j].equals(languages[k])) {
						score = 6-j;
						break;
					}
				}
				total += score * preference[k];
				
			}
			if(total >= max) {
				max = total;
				answer = i;
			}
		}
        return table2[answer][0];
    }
}

 

split 함수를 사용하여 " "을 기준으로 문자열을 나누고 2차원 배열에 넣었습니다.

2차원 배열의 table2와 languages를 비교하면서 같으면 점수를 부여했습니다.

그리고 같은 점수면 이름순으로 앞순서의 언어를 출력하기 위해 내림차순을 했습니다.

 

 

정확성 테스트를 보니 제 코드는 완벽히 실패했습니다....

 

---

이외 풀이

 

import java.util.*;
class Solution {
    public String solution(String[] table, String[] languages, int[] preference) {
        String answer = "";
        int score=-1;
        for(String str : table){
            String[] t = str.split(" ");
            String tname = t[0];
            int tscore = 0;
            for(int i=0;i<languages.length;i++){
                int idx = Arrays.asList(t).indexOf(languages[i]);
                if(idx>-1) tscore+=preference[i]*(6-idx);
            }
            if(score ==tscore && answer.compareTo(tname)>0) answer=tname;
            if(score<tscore){
                score =tscore;
                answer= tname;
            }
        }
        return answer;
    }
}

for문 두개로 어떻게 이렇게 짧게 풀 수 있는지 경의롭습니다...

 

Arrays의 asList로 ArrayList 형태로 변환합니다.

그리고 languages의 각 요소들과 비교하면 인덱스를 추출합니다.

해당 값이 존재하면 식에 따라 점수를 구합니다.

 

 

https://programmers.co.kr/learn/challenges

코딩테스트 연습