[프로그래머스] 다른 사람의 풀이

문제 : 프로그래머스 참조

프로그래머스 이전 주차의 문제들을 다 풀고 이번에는 처음으로 제대로 주차 수 맞춰서 풀게 되었다. 4주차 문제 GO!

[문제]
개발자가 사용하는 언어와 언어 선호도를 입력하면 그에 맞는 직업군을 추천해주는 알고리즘을 개발하려고 합니다.
아래 표는 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" 중 하나입니다.

[입력]

[설명]


문제 이해는 별로 어렵지는 않은데 문자열 Parsing이 어렵다.
(운영체제의 플젝1의 악몽ㄷㄷ명령어 옵션마다 parsing해서 stacking하는 거 세상에서 제일 끔찍했어...)

c++의 string은 c의 char와 조금 다르다. 아무래도 string의 method만 따로 한 번 정리해야할 거 같다.

여기서는 sstream을 따로 쓰지 않고 string의 find와 substr 메소드만을 활용해서 parsing해주었다.

  • previous와 current 변수를 이용해서 string의 indexing을 해주었다.
  • str.find(" ", previous) : previous 인덱스 이후에 있는 공백의 인덱스를 리턴
  • str.substr(previous, current - previous) : previous 인덱스부터 (current - previous)만큼 substring한다. 헷갈리면 배열을 그려보자!ㅎㅎ

이렇게 문자열이 많이 등장하는 경우에는 대다수 자료구조로 HashMap을 사용한다..
(으으 C++, Python, Java, Javascript 등등 주요 프로그래밍 언어를 매번 프로젝트 때마다 달리 쓰니 헷갈린다..특히 메소드...ㅂㄷㅂㄷ...Java에서는 HashMap이고, C++에서는 unordered_map이다. 그런데 자료구조 이론상으로는 HashMap이 더 적절한 거 같아서 위와 같이 표기하겠다...)

사설은 멈추고 문제를 조금 더 자세히 분석하면,
파라미터는 table, languages, preference 총 세 개의 vector가 주어진다. vector은 다음과 같이 활용한다.

  • table은 parsing해서 language별로 hashmap의 점수를 반영한다.
  • languages + preference = hashmap

    🌟참고로 JAVA에서는 HashMap에 대한 삽입 메소드로 put을 쓰지만 cpp에는 operator []가 존재해서 insert or update가 가능하다.

이외에도 구조체 vector를 사용하여 구조체를 score 내림차순, job 오름차순으로 정렬하여 첫 번째 job의 이름을 리턴한다. 이를 위해 sort의 parameter함수로 cmp 비교 함수를 생성한다. 이 때 리턴 값은 반드시 bool 타입이어야 한다. (int로 해보니 안 먹히더라...이 말이야)

#include <string>
#include <vector>
#include <unordered_map>
#include <algorithm>
#include <iostream>
using namespace std;

struct programmer{
    int score;
    string job;
};

unordered_map<string, int> prefer;
vector<programmer> v;
bool cmp(const struct programmer &a, const struct programmer &b);
string solution(vector<string> table, vector<string> languages, vector<int> preference) {
    for(int i = 0;i < languages.size();i++) {
        prefer[languages[i]] = preference[i];
    }

    for(auto str : table) {
        int current = str.find(" ");
        int score = 0;
        int i = 5;
        string job = str.substr(0, current);
        int previous = current + 1;
        current = str.find(" ", previous);
        while(i) {
            string lang = str.substr(previous, current - previous);
            score += prefer[lang] * i;
            previous = current + 1;
            current = str.find(" ", previous);
            i--;
        }
        v.push_back({score,job});
    }
    sort(v.begin(), v.end(), cmp);
    return v[0].job;
}

bool cmp(const struct programmer &a, const struct programmer &b) {
    if(a.score > b.score) return true;
    else if(a.score == b.score) {
        return a.job < b.job;
    }
    else return false;
}

좋은 웹페이지 즐겨찾기