리코딩 : 알고리즘

프로그래머스 - 직업군 추천하기

BreezeBm 2021. 10. 7. 13:53

1. 문제

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

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

점수SICONTENTSHARDWAREPORTALGAME

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"

2. 나의풀이

function solution(table, languages, preference) {
  // 점수 배열
  const point = [5, 4, 3, 2, 1];
  // table을 객체로 정렬
  let table_obj = {};
    
  // 반복문을 통해서 배열의 0번째 값을 키로 가지고, 직업군 별로 계산한 값을 value로 가진다.  
  for(let i = 0; i < table.length; i ++) {
    const str = table[i].split(" "); // ["SI", "JAVA", "JAVASCRIPT", "SQL", "PYTHON", "C#"]
    if(!table_obj[str[0]]) table_obj[str[0]] = 0; // { "SI": 0 }
    
    for (let j = 0; j < languages.length; j ++) {
      // 직업군 언어가 포함되어 있는지 확인
      let include = str.includes(languages[j]);
      // 직업군 언어의 인덱스를 가져온다.
      const strIdx = str.indexOf(languages[j]);
      // 인덱스에서 -1을 해야, 해당하는 점수와 일치
      const idxPoint = point[strIdx - 1];
      
      // 포함되어 있을 때만, 값을 구해서 객체의 값으로 넣어준다.
      if(include) table_obj[str[0]] += (idxPoint * preference[j]) 
    }
  }
  
  //[["SI",55],["CONTENTS",53],["HARDWARE",19],["PORTAL",55],["GAME",22]]
  const ordered_obj = Object.entries(table_obj).sort().sort(([a, b], [c, d]) => d - b)
  return ordered_obj[0][0];
}

- 객체로 직업별로 정리가 되어있으면 좋겠다는 생각이 들었다. 

- table_obj를 선언한다.

- 반복문을 통해서 "SI JAVA JAVASCRIPT SQL PYTHON C#" 문자열을 " "을 기준으로 나눈다.

  ( ["SI", "JAVA", "JAVASCRIPT", "SQL", "PYTHON", "C#"] )

- 나누어진 문자열의 0번째 인덱스는 직군이다. 직군을 key로 가지는 객체를 만들기 위해 table_obj에 넣어준다.

 

- 두번째 반복문을 통해서 직군안에 개발자가 사용하는 언어가 있는지를 순회하면서 확인한다.- 그리고 ["SI", "JAVA", "JAVASCRIPT", "SQL", "PYTHON", "C#"] 배열안에 몇번째에 있는지를 가져온다.- 그리고 언어의 point를 가져온다.- 만약 포함이 되어있으면 직군에 일치하는 키에 점수를 계산한 값을 넣어준다.

 

- 마지막에 객체를 정렬을 한다. (이부분이 힘들었다.)

- 객체를 정렬할 수 있는 방법을 찾다보니, 많은 코드를 작성해야 했다.- Object.entries()는 객체의 키와 값을 배열로 만들어준다.     [ ["SI",55], ["CONTENTS",53], ["HARDWARE",19], ["PORTAL",55], ["GAME",22] ]
- 만들어진 배열을 우선 직군 기준으로 정렬을 하고, 점수를 기준으로 정렬을 한다.

  [ [ 'PORTAL', 55 ], [ 'SI', 55 ], [ 'CONTENTS', 53 ], [ 'GAME', 22 ], [ 'HARDWARE', 19 ] ]

- 정렬된 배열의 0번째 인덱스의 0번째 값을 반환하면 된다.

 공부... 공부....