33-kokeunho

[kokeunho]

🔗 문제 링크

[PCCP 기출문제] 4번 수식 복원
https://school.programmers.co.kr/learn/courses/30/lessons/340210

✔️ 소요된 시간

3h? (풀이는 쉬운데 구현 역량이 좀 부족해서...)

✨ 수도 코드

해당 문제 풀이는 간단합니다.
입력 받은 식에서 가능한 진법 범위를 찾아낸 후
식을 성립 시키는 진법들만 추려냅니다.
그리고 해당 진법들로 X를 구하고 출력을 구하면 됩니다.

구현 역량이 부족해서 헤맨 문제였습니다.
다른 분들은 어렵지 않게 풀 수 있을 것이라고 생각합니다.

더 자세한 풀이는 블로그 포스팅으로 작성해봤습니다.
https://velog.io/@kokangsik65/PCCP-기출-4번-구현을-하지-못하는-나에게

코드 리뷰든 포스트 리뷰든 달게 받겠습니다!

📚 새롭게 알게된 내용

[kangrae-jo]

약 2시간동안의 gpt와 claude를 고문한 끝에 정답을 얻어냈습니다.

처음에 한 30-40분은 문제를 잘 못 이해해서 고생을 좀 했네요.
저기 나와있는 수가 10진수인줄 알았습니다.
그러나 입력으로 주어지는 수들은 모두 n진수 더라구요.
14 + 3 = 17을 보고 한참 멍때렸네요.

문제를 푸는 방식은 아래와 같습니다.

1. 식 tokenize 하면서 X가 포함된 식 분리
2. 2진법 부터 9진법 까지 적절히 적용시켜보기( exps를 하나씩 꺼내가며 식이 맞는지확인)
3. 후보에 있는 진법들 적용시 값이 달라지는 녀석은 ?로 넣고, 안달라지는 녀석은 그 값을 넣습니다.

이때 정답 배열에 넣을때도 진법을 유지해야해요.
따라서 증감 연산은
N진법에서 10진법으로 바꾸고,
계산하고,
10진법에서 N진법으로 바꿔야 합니다.
코드에 정의되어있는 + - 는 10진법 기반 계산이기 때문이죠.

그리고 또 거의 1시간을 쓴 부분이 있는데, 이거 하나 고쳐주니까 정답이더라구요.
그건 바로 모든 자릿수를 확인하면서 최소 진법? 을 찾는 거에요.
진법이 적용이 되냐 안되냐로 판단할 수 있을 거라 생각했는데, 아니었나봐요...

오랜만에 매우매우 어렵고 힘들었네요.

근호님 코드도 많이 도움이 되었습니다. 감사합니다.

CPP CODE

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

using namespace std;

int nToTen(const string& s, int base, bool& ok) {
    int val = 0;
    ok = true;
    for (char ch : s) {
        if (ch < '0' || ch > '9') { 
            ok = false; 
            return 0; 
        }
        
        int d = ch - '0';
        if (d >= base) { 
            ok = false; 
            return 0; 
        }
        val = val * base + d;
    }
    return val;
}

string tenToN(int val, int base) {
    if (val == 0) return "0";
    string result = "";
    while (val > 0) {
        result = char('0' + val % base) + result;
        val /= base;
    }
    return result;
}

struct Exp {
    string a, s, b, c;
    Exp(string a_, string s_, string b_, string c_) : a(a_), s(s_), b(b_), c(c_) {}
    
    bool isCollectAry(int ary) {
        bool okA, okB, okC;
        int A = nToTen(a, ary, okA);
        int B = nToTen(b, ary, okB);
        int C = nToTen(c, ary, okC);
        if (!okA || !okB || !okC) return false;
        if (s == "+") return (A + B) == C;
        else          return (A - B) == C;
    }
    
    string getResult(int ary) {
        bool okA, okB;
        int A = nToTen(a, ary, okA);
        int B = nToTen(b, ary, okB);
        if (!okA || !okB) return "";
        
        int result;
        if (s == "+") result = A + B;
        else          result = A - B;
        
        return tenToN(result, ary);
    }
    
    string toString(string result) {
        return a + " " + s + " " +  b + " = " + result;
    }
    
    // 식에 포함된 모든 숫자에서 가장 큰 자릿수 찾기
    int getMaxDigit() {
        int maxDigit = 0;
        for (char ch : a) {
            if (ch >= '0' && ch <= '9') maxDigit = max(maxDigit, ch - '0');
        }
        for (char ch : b) {
            if (ch >= '0' && ch <= '9') maxDigit = max(maxDigit, ch - '0');
        }
        if (c == "X") return maxDigit;
        for (char ch : c) {
            if (ch >= '0' && ch <= '9') maxDigit = max(maxDigit, ch - '0');
        }
        return maxDigit;
    }
};

vector<string> solution(vector<string> expressions) {
    vector<Exp> xs, exps;
    for (int i = 0; i < expressions.size(); i++) {
        stringstream ss(expressions[i]);
        string a, sign, b, equal, c;
        ss >> a >> sign >> b >> equal >> c;
        
        if (c == "X") xs.push_back(Exp(a, sign, b, c));
        else exps.push_back(Exp(a, sign, b, c));
    }
    
    // 모든 식에서 최대 자릿수 찾기
    int maxDigit = 0;
    for (Exp exp : exps) {
        maxDigit = max(maxDigit, exp.getMaxDigit());
    }
    for (Exp exp : xs) {
        maxDigit = max(maxDigit, exp.getMaxDigit());
    }
    
    // 최대 자릿수 + 1진법부터 시작
    vector<int> arys;
    for (int ary = maxDigit + 1; ary <= 9; ary++) {
        bool flag = true;
        for (Exp exp : exps) {
            if (!exp.isCollectAry(ary)) {
                flag = false;
                break;
            }    
        }
        if (flag) arys.push_back(ary);
    }
    
    vector<string> answer;
    for (Exp exp : xs) {
        vector<string> results;
        for (int ary : arys) {
            string result = exp.getResult(ary);
            if (!result.empty()) {
                results.push_back(result);
            }
        }
        
        bool allSame = true;
        if (results.size() > 1) {
            for (int i = 1; i < results.size(); i++) {
                if (results[i] != results[0]) {
                    allSame = false;
                    break;
                }
            }
        }
        
        if (allSame && !results.empty()) answer.push_back(exp.toString(results[0]));
        else answer.push_back(exp.toString("?"));
    }
    
    return answer;
}