본문 바로가기
Kakao Coding Test/Level2

방금그곡 C++ 2018 KAKAO Blind Recruitment

by HenryNoh 2021. 4. 8.

문제 설명

방금그곡

라디오를 자주 듣는 네오는 라디오에서 방금 나왔던 음악이 무슨 음악인지 궁금해질 때가 많다. 그럴 때 네오는 다음 포털의 '방금그곡' 서비스를 이용하곤 한다. 방금그곡에서는 TV, 라디오 등에서 나온 음악에 관해 제목 등의 정보를 제공하는 서비스이다.

네오는 자신이 기억한 멜로디를 가지고 방금그곡을 이용해 음악을 찾는다. 그런데 라디오 방송에서는 한 음악을 반복해서 재생할 때도 있어서 네오가 기억하고 있는 멜로디는 음악 끝부분과 처음 부분이 이어서 재생된 멜로디일 수도 있다. 반대로, 한 음악을 중간에 끊을 경우 원본 음악에는 네오가 기억한 멜로디가 들어있다 해도 그 곡이 네오가 들은 곡이 아닐 수도 있다. 그렇기 때문에 네오는 기억한 멜로디를 재생 시간과 제공된 악보를 직접 보면서 비교하려고 한다. 다음과 같은 가정을 할 때 네오가 찾으려는 음악의 제목을 구하여라.

  • 방금그곡 서비스에서는 음악 제목, 재생이 시작되고 끝난 시각, 악보를 제공한다.
  • 네오가 기억한 멜로디와 악보에 사용되는 음은 C, C#, D, D#, E, F, F#, G, G#, A, A#, B 12개이다.
  • 각 음은 1분에 1개씩 재생된다. 음악은 반드시 처음부터 재생되며 음악 길이보다 재생된 시간이 길 때는 음악이 끊김 없이 처음부터 반복해서 재생된다. 음악 길이보다 재생된 시간이 짧을 때는 처음부터 재생 시간만큼만 재생된다.
  • 음악이 00:00를 넘겨서까지 재생되는 일은 없다.
  • 조건이 일치하는 음악이 여러 개일 때에는 라디오에서 재생된 시간이 제일 긴 음악 제목을 반환한다. 재생된 시간도 같을 경우 먼저 입력된 음악 제목을 반환한다.
  • 조건이 일치하는 음악이 없을 때에는 “(None)”을 반환한다.

입력 형식

입력으로 네오가 기억한 멜로디를 담은 문자열 m과 방송된 곡의 정보를 담고 있는 배열 musicinfos가 주어진다.

  • m은 음 1개 이상 1439개 이하로 구성되어 있다.
  • musicinfos는 100개 이하의 곡 정보를 담고 있는 배열로, 각각의 곡 정보는 음악이 시작한 시각, 끝난 시각, 음악 제목, 악보 정보가 ','로 구분된 문자열이다.
    • 음악의 시작 시각과 끝난 시각은 24시간 HH:MM 형식이다.
    • 음악 제목은 ',' 이외의 출력 가능한 문자로 표현된 길이 1 이상 64 이하의 문자열이다.
    • 악보 정보는 음 1개 이상 1439개 이하로 구성되어 있다.

출력 형식

조건과 일치하는 음악 제목을 출력한다.

입출력 예

m musicinfos answer
"ABCDEFG" ["12:00,12:14,HELLO,CDEFGAB", "13:00,13:05,WORLD,ABCDEF"] "HELLO"
"CC#BCC#BCC#BCC#B" ["03:00,03:30,FOO,CC#B", "04:00,04:08,BAR,CC#BCC#BCC#B"] "FOO"
"ABC" ["12:00,12:14,HELLO,C#DEFGAB", "13:00,13:05,WORLD,ABCDEF"] "WORLD"

설명

첫 번째 예시에서 HELLO는 길이가 7분이지만 12:00부터 12:14까지 재생되었으므로 실제로 CDEFGABCDEFGAB로 재생되었고, 이 중에 기억한 멜로디인 ABCDEFG가 들어있다.
세 번째 예시에서 HELLO는 C#DEFGABC#DEFGAB로, WORLD는 ABCDE로 재생되었다. HELLO 안에 있는 ABC#은 기억한 멜로디인 ABC와 일치하지 않고, WORLD 안에 있는 ABC가 기억한 멜로디와 일치한다.

구현

문제 자체의 난이도는 크게 높지 않은 편에 속한다. 하지만 항상 카카오 문제들이 그러하듯 문자열 처리에 대한 부분을 쉽고 빠르게 해결해낼 수 있는가, 문제에 주어진 조건들을 잘 파악하고 있는가 등을 중점적으로 보는 문제인 것 같다.

해당 문제는 크게 3가지 파트로 구성된다.

  1. 입력으로 들어온 문자열에 대한 처리
  2. #에 대한 처리와 멜로디를 재생시간만큼 표현하는 것
  3. 예외처리

3가지 다 코딩을 어느 정도만 해왔다고 하더라도 쉽게 파악할 수 있는 부분이기도 할뿐더러 그렇게 어려운 부분도 없다. 따라서 얼마나 적절한 자료구조를 사용하고 얼마나 빨리 문제를 파악하고 해결하는지에 대한 부분을 검사하는 문제인 것 같다.


우선 입력으로 들어온 musicinfos를 ", "로 나누어 infos 벡터에 새롭게 저장해준다.

vector<vector<string>> infos(musicinfos.size());
for (int i = 0; i < musicinfos.size(); i++)
{
  stringstream ss(musicinfos[i]);
  string temp;
  while (getline(ss, temp, ','))
 	 infos[i].push_back(temp);
}

다만 이때 m과 musicinfos에서 C#, D#, F#, G#, A# 같은 경우에는 문자를 두 개씩 처리해줘야 하기 때문에 각 알파벳에 10을 더하여 새롭게 저장해주었다.

for (int i = 0; i < infos.size(); i++)
{
  for (int j = 0; j < infos[i][3].size();)
  {
    if (infos[i][3][j + 1] == '#')
    {
      infos[i][3][j] += 10;
      infos[i][3].erase(infos[i][3].begin() + j + 1);
    }
    else
   	 j++;
  }
}
for (int i = 0; i < m.size();)
{
  if (m[i + 1] == '#')
  {
    m[i] += 10;
    m.erase(m.begin() + i + 1);
  }
  else
  	i++;
}

다음은 infos vector에 들어간 시간을 처리하기 위하여 ":"를 구분자로 나누어준 후에 시간을 계산하여 times vector에 저장해주었다.

for (int i = 0; i < infos.size(); i++)
{
  stringstream ss1(infos[i][0]);
  stringstream ss2(infos[i][1]);
  int t1 = 0;
  int t2 = 0;
  string temp1;
  string temp2;
  while (getline(ss1, temp1, ':'))
  	t1 = (t1 * 60) + stoi(temp1);
  while (getline(ss2, temp2, ':'))
  	t2 = (t2 * 60) + stoi(temp2);
  times.push_back(abs(t2 - t1));
}

입력으로 들어온 music infos에 대한 가공처리가 끝난 후에는 musicinfos의 멜로디를 시간만큼 따로 저장해주어야 한다.

vector<string> temp(infos.size());
for (int i = 0; i < infos.size(); i++)
{
  for (int j = 0; j < times[i]; j++)
  {
  	temp[i].push_back(infos[i][3][j % (infos[i][3].size())]);
  }
}

마지막으로 길이만큼 멜로디가 재생된 것을 저장해둔 temp vector를 m으로 탐색하며 두 가지 예외조건을 만족시키도록 결과를 도출해낸다.

  1. 조건이 일치하는 음악이 여러 개일 때에는 라디오에서 재생된 시간이 제일 긴 음악 제목을 반환한다. 재생된 시간도 같을 경우 먼저 입력된 음악 제목을 반환한다.
  2. 조건이 일치하는 음악이 없을 때에는 “(None)”을 반환한다.
int max = 0;
for (int i = 0; i < temp.size(); i++)
{
  if (temp[i].find(m) != string::npos && times[i] > max)
  {
    max = times[i];
    answer = infos[i][2];
  }
}
if (answer == "")
	answer = "(None)";
return answer;

전체 코드는 아래에서 볼 수 있다.

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

using namespace std;

string solution(string m, vector<string> musicinfos)
{
    string answer = "";
    vector<vector<string>> infos(musicinfos.size());
    for (int i = 0; i < musicinfos.size(); i++)
    {
        stringstream ss(musicinfos[i]);
        string temp;
        while (getline(ss, temp, ','))
            infos[i].push_back(temp);
    }
    for (int i = 0; i < infos.size(); i++)
    {
        for (int j = 0; j < infos[i][3].size();)
        {
            if (infos[i][3][j + 1] == '#')
            {
                infos[i][3][j] += 10;
                infos[i][3].erase(infos[i][3].begin() + j + 1);
            }
            else
                j++;
        }
    }
    for (int i = 0; i < m.size();)
    {
        if (m[i + 1] == '#')
        {
            m[i] += 10;
            m.erase(m.begin() + i + 1);
        }
        else
            i++;
    }
    vector<int> times;
    for (int i = 0; i < infos.size(); i++)
    {
        stringstream ss1(infos[i][0]);
        stringstream ss2(infos[i][1]);
        int t1 = 0;
        int t2 = 0;
        string temp1;
        string temp2;
        while (getline(ss1, temp1, ':'))
            t1 = (t1 * 60) + stoi(temp1);
        while (getline(ss2, temp2, ':'))
            t2 = (t2 * 60) + stoi(temp2);
        times.push_back(abs(t2 - t1));
    }
    vector<string> temp(infos.size());
    for (int i = 0; i < infos.size(); i++)
    {
        for (int j = 0; j < times[i]; j++)
        {
            temp[i].push_back(infos[i][3][j % (infos[i][3].size())]);
        }
    }
    int max = 0;
    for (int i = 0; i < temp.size(); i++)
    {
        if (temp[i].find(m) != string::npos && times[i] > max)
        {
            max = times[i];
            answer = infos[i][2];
        }
    }
    if (answer == "")
        answer = "(None)";
    return answer;
}

댓글