[프로그래머스/C++] 직업군 추천하기

[프로그래머스/C++] 직업군 추천하기

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 해주세요.

2. 제한사항

• 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" 중 하나입니다.

3. 풀이

• 문제에서 시키는대로 하면 되는 구현 문제
• 정답을 사전 순으로 가장 빠른 직업 군 하나만 출력할 것이기 때문에 table을 미리 사전 순으로 정렬해주었다.
• 공백을 기준으로 단어 하나를 가져오는 getWord 함수를 만들어 사용하였다.

string getWord(string s, int &index) {
    string word = "";
    
    while (true) {
        if (index >= s.size() || s[index] == ' ') break;
        word += s[index++];
    }
    
    return word;
}

• 1차원 string 배열인 table을 2차원 배열로 바꾸어 주었다.
  • 2차원 배열로 바꿔야 인덱스로 점수 계산하기 편할 것 같아서 바꾸었다.
  • 안 바꾸어도 풀 수 있을 것 같다.
• 2차원 배열이 끝나면 languages배열을 돌며 언어 선호도 x 직업군 언어 점수를 계산해서 계속 더해준다.
• 최댓값을 구하고 최댓값 위치에 있는 직업군을 출력한다.
  • 사전 순으로 table을 정렬하였기 때문에 가장 먼저 나오는 최댓값을 리턴하면 된다.

4. 처음 코드와 달라진 점

• null
• 좀 더 빠른 방법으로 만들고 싶다는 생각은 있다.

5. 코드

#include <iostream>
#include <string>
#include <vector>
#include <algorithm>

using namespace std;

string getWord(string s, int &index) {
    string word = "";
    
    while (true) {
        if (index >= s.size() || s[index] == ' ') break;
        word += s[index++];
    }
    
    return word;
}

string solution(vector<string> table, vector<string> languages, vector<int> preference) {
    string answer = "";
    sort(table.begin(), table.end());

    vector<vector<string>> score(table.size());
    for (int i = 0; i < table.size(); ++i) {
        int index = 0;
        string temp = getWord(table[i], index);
        index++;
        
        while (index < table[i].size()) {
            temp = getWord(table[i], index);
            index++;
            
            score[i].push_back(temp);
        }
    }
    
    vector<int> result(table.size(), 0);
    for (int i = 0; i < languages.size(); ++i) {
        for (int j = 0; j < score.size(); ++j) {
            for (int k = 0; k < score[j].size(); ++k) {
                if (score[j][k] == languages[i]) {
                    result[j] += (5 - k) * preference[i];
                }
            }
        }
    }
    
    int maxValue = 0;
    for (int i = 0; i < result.size(); ++i) {
        maxValue = max(maxValue, result[i]);
    }
    
    for (int i = 0; i < result.size(); ++i) {
        if (result[i] == maxValue) {
            int index = 0;
            answer = getWord(table[i], index);
            break;
        }
    }
    return answer;
}