JUNGOL 실력키우기 1239 : 비밀편지

병현이는 지은이에게 문자 A, B, C, D, E, F, G, H 로 쓰여진 편지를 날마다 보내는데, 

컴퓨터로 보내는 비밀편지로, 한 문자마다 0 또는 1인 숫자 여섯 개를 사용하여 보낸다. 

둘 사이의 약속은 다음과 같다.

A 000000

B 001111

C 010011

D 011100

E 100110

F 101001

G 110101

H 111010

병현이가 어느 날 001111000000011100 을 보내면 지은이는 이것을 BAD로 이해하게 된다. 그런데 둘 사이에 약속이 잘 만들어져 있기 때문에, 통신에 문제가 생겨서 한 문자를 표시하는 여섯 숫자 중 어느 한 숫자만 틀리게 오는 경우, 지은이는 원래 보내려는 문자를 알아 낼 수가 있다.

 

예를 들어 지은이가 000100을 받았을 때, a와 숫자 한자만 다르고, 다른 문자들과는 각각 숫자 두 자 이상이 다르므로 지은이는 이것이 A라고 알아보게 된다.

다만 111111과 같이 모든 문자의 표현과 숫자 두 자 이상이 다른 경우에는 무슨 문자인지 알 수가 없게 된다. 예를 들어 지은이가 011111000000111111000000111111 을 받았을 때, BA 다음에 알아 볼 수 없는 문자가 나오는데. 이 경우 이런 것이 처음 나오는 문자의 위치인 3을 출력한다.

 

지은이가 받은 편지를 보고 문자들을 알아내어 출력하거나, 모르는 문자가 있는 경우, 이것이 처음 나오는 위치를 출력하는 프로그램을 작성하시오.

 

첫줄에는 보낸 문자의 개수(10개 보다 작다.)가 입력된다.
다음 줄에는 문자의 개수의 여섯 배 만큼의 숫자 입력이 주어진다.

 

주어진 입력에서 지은이가 이해한 문자들을 출력하거나 모르는 문자가 나오는 경우 그런 것이 처음 나오는 위치를 출력한다.

[Copy] 3 001111000000011100

[Copy] BAD

[Copy] 5 011111000000111111000000111111

[Copy] 3

---

처음 문제 읽고 어떻게 풀어야 할지 한번에 감이 오질 않아서 그냥 의식의 흐름대로 코드를 작성했는데 어떻게든 해결은 됐습니다.

그런데 코드가 영 보기 좋지가 않네요. 일단 다른 분들은 어떻게 해결했는지 검색해 보고

괜찮은 코드가 있으면 분석해 보고 다른 버전으로 해결을 해 봐야할 것 같네요.

P.S.

 다른 블로그에서 코드를 분석해 본 결과 과정은 거의 비슷한데 클래스를 이용했을 때 상당히 보기 깔끔해지네요.

거의 제가 추구하는 읽히는 코드가 되는 것 같습니다. 추가로 약간의 속도 향상이 있는 듯 합니다.

#include <iostream>
#include <string>
#include <map>
 
using namespace std;
 
// 문자의 끝을 의미하도록 할 글자를 알아보기 쉽게 정의한다.
#define ENDCODE '\0' 
 
char FindCorrectWord(string& word);
 
// 코드를 문자로 변환하기 위한 데이터
map<string, char> codeToWord
{
    { "000000", 'A' },
    { "001111", 'B' },
    { "010011", 'C' },
    { "011100", 'D' },
    { "100110", 'E' },
    { "101001", 'F' },
    { "110101", 'G' },
    { "111010", 'H' }
};
 
// 코드가 정확하지 않은 경우 제대로 된 문자를 찾기 위한 데이터
map<char, string> wordToCode
{
    {'A', "000000"},
    { 'B', "001111" },
    { 'C', "010011" },
    { 'D', "011100" },
    { 'E', "100110" },
    { 'F', "101001" },
    { 'G', "110101" },
    { 'H', "111010" }
};
 
int main(void)
{
    int n;
    string letter;
    string result;
 
    cin >> n;
    cin >> letter;
 
    for (int i = 0; i < n; ++i)
    {
        // 문자열을 6개씩 잘라낸다.
        string word = letter.substr(i * 6, 6);
        
        // map에 대입한 결과가 존재하면 그것을 결과에 추가한다.
        if (codeToWord[word] != NULL)
        {
            result.push_back(codeToWord[word]);
        }
        else
        {
            // 결과가 존재하지 않는 경우에는 코드가 틀린 경우 이므로 검색하여 찾는다.
            // 만약 그 결과가 존재하지 않는경우(모두 두글자 이상 틀린경우)
            // 현재 문자의 순서(처음으로 틀린문자가 등장하는 위치)를 출력하고 종료한다.
            result.push_back(FindCorrectWord(word));
            if (result.back() == ENDCODE)
            {
                cout << i +1<< endl;
                return 0;
            }
        }
    }
    cout << result << endl;
}
 
// 틀린 코드에서 원래 의도된 글자를 찾는 함수
char FindCorrectWord(string& word)
{
    int counter;
    char result;
 
    // 문제에서 주어진 'A' 부터 'H' 까지의 코드를 검사한다.
    for (char i = 'A'; i <= 'H'; ++i)
    {
        counter = 0;
        string cur = wordToCode[i];
        for (int j = 0; j < 6; ++j)
        {
            // 코드가 다른 경우 카운트를 1 증가시킨다.
            if (cur[j] != word[j])
                counter++;
            // 카운트가 2이상인 경우 더 이상 검사할 필요가 없으므로
            // 종료하고 다음 문자의 코드를 검사한다.
            if (counter >= 2)
                break;
        }
        // 만약 카운트가 1인 경우 의도된 글자 이므로 그것을 리턴한다.
        if (counter == 1)
        {
            return i;
        }
    }
    // 만약 모든 글자를 검사하였으나 모든 문자가 2개 이상 틀린 경우
    // 끝을 의미할 글자를 리턴한다.
    return ENDCODE;
}

#include <iostream>
#include <string>
 
using namespace std;
 
#define NONE '\0'
 
class Code
{
private:
    char word;
    string code;
 
public:
    Code(char word, string code)
        : word(word), code(code) {}
 
    // 코드가 정확하게 일치하는 경우 문자를 리턴한다.
    char GetCorrectWord(string& otherCode)
    {
        if (code == otherCode)
            return word;
        return NONE;
    }
 
    // 코드를 한글자씩 비교하며 한개만 틀린 경우 문자를 리턴한다.
    char GetSimilarWord(string& otherCode)
    {
        int cnt = 0;
 
        for (int i = 0; i < 6; ++i)
        {
            if (code[i] != otherCode[i])
                cnt++;
            if (cnt > 1)
                return NONE;
        }
        return word;
    }
};
 
Code codes[] = 
{
    { 'A', "000000" },
    { 'B', "001111" },
    { 'C', "010011" },
    { 'D', "011100" },
    { 'E', "100110" },
    { 'F', "101001" },
    { 'G', "110101" },
    { 'H', "111010" }
};
 
char GetCorrectWord(string& codeSet);
char GetSimilarWord(string& codeSet);
bool CanUnderstand(char word);
 
int main(void)
{
    int n;
    string letter;
    string result;
 
    cin >> n;
    cin >> letter;
 
    for (int i = 0; i < n; ++i)
    {
        // 코드를 6개씩 잘라서 받아온다.
        string codeSet = letter.substr(i * 6, 6);
        // 코드를 이용하여 문자를 해석한다.
        char word = GetCorrectWord(codeSet);
        
        // 해석이 된 경우 저장한다.
        if (CanUnderstand(word))
        {
            result.push_back(word);
        }
        // 해석이 안된 경우 위치를 출력하고 종료한다.
        else
        {
            cout << i + 1 << endl;
            exit(0);
        }
    }
    // 해석이 완료된 경우 결과를 출력한다.
    cout << result << endl;
}
 
// 코드가 정확하게 일치하는 문자를 리턴한다.
char GetCorrectWord(string& codeSet)
{
    char word;
 
    for (int i = 0; i < 8; ++i)
    {
        word = codes[i].GetCorrectWord(codeSet);
        if (word != NONE)
            return word;
    }
    return GetSimilarWord(codeSet);
}
 
// 코드가 비슷한(1개다른) 문자를 리턴한다.
char GetSimilarWord(string& codeSet)
{
    char word;
 
    for (int i = 0; i < 8; ++i)
    {
        word = codes[i].GetSimilarWord(codeSet);
        if (word != NONE)
            return word;
    }
    return NONE;
}
 
// 문자가 존재하는지 판단한다.
bool CanUnderstand(char word)
{
    if (word == NONE)
        return false;
    return true;
}
 

JUNGOL) 문제은행) 실력키우기) 비밀편지