자바(백준) 23629 이 얼마나 끔찍하고 무시무시한 수식이니

오답 노트 & 새로 알게 된 점

이 문제를 푸는데 6시간이 걸린 듯 하다.

총 23트 끝에 문제를 맞게 됐다. 진짜 끔찍하고 무시무시했던 문제였다.

 

문제를 정리해보면

1. 숫자를 문자열로 풀어서 입력을 줌 

2. 문자열을 숫자로 변환하여 계산을 함

3. 계산된 값을 다시 문자열로 풀어야 함

 

처음에 문자열을 완탐하면서 연산을 배열에 넣어줬다.

왜?

연산의 갯수가 숫자문자열의 개수보다 많다면 잘못된 수식임을 판별하기 위해서이다.

 

StringTokenizer를 이용하여 연산자를 구분자로 문자열을 파싱하면서,

정답을 출력하기 위해 Stringbuilder에 값을 넣어주면서, 숫자를 변환하여 덱에 넣어주었다.

 

덱에 넣어준 이유는 값을 계산하기 위해서이다

-> 근데 왜 덱을 사용했을까?

-> 계산하기 위해서는 두 개의 숫자가 필요하다. 또한, 앞에서부터 계산한다고 했으므로

두 개의 숫자를 꺼낸 뒤에, 계산한 값을 다시 제일 앞에 넣어주면서 순차적으로 계산하기 위함이다.

 

 

나의 코드를 그림으로 정리해봤다.

주석에 있는 숫자 순서대로 이해하면 된다.

다만, 덱에 있는 계산식은 코드를 보면서 이해하는게 좋다.

덱의 원소의 개수가 하나가 될 때까지 계속해서 계산해나간다.

 

아래는 문제 예제 4번이다.

 

---

코드

import java.io.*;
import java.util.*;
 
public class Main {
    static Long atoi(String str) {
        return Long.parseLong(str);
    }
 
    static HashMap<String, String> hm = new HashMap<>();
    static String s;
    static ArrayList<Character> op = new ArrayList<>();
    static StringBuilder sb = new StringBuilder();
    static Deque<Long> dq = new ArrayDeque<>();
    public static void main(String[] args) throws IOException {
        input();
        pro();
 
        System.out.println(sb);
    }
 
    static void pro() {
        StringTokenizer st = new StringTokenizer(s, "+-x/=");
 
        if(st.countTokens() < op.size()){
            sb.append("Madness!");
            return;
        }
 
        int opIdx = 0;
        long sum = 0L;
 
        //1
        while (st.hasMoreTokens()) {
            String rel = stringToNum(st.nextToken());
 
            if(rel == null){
                sb.append("Madness!");
                return;
            }
 
            sb.append(rel).append(op.get(opIdx++));
            dq.offer(atoi(rel));
        }
 
        sb.append("\n");
 
        //2
        opIdx = 0;
        while (dq.size() >= 2) {
            long l1 = dq.poll();
            long l2 = dq.poll();
 
            long rel = calcul(l1, l2, op.get(opIdx));
            opIdx++;
            dq.offerFirst(rel);
        }
 
 
        //3
        long l = dq.poll();
        sb.append(numToString(l));
 
    }
 
    static String numToString(Long l) {
        String strL = String.valueOf(l);
        String rel = "";
        for (int i = 0; i < strL.length(); i++) {
            if(strL.charAt(i) == '+' || strL.charAt(i) == '-' || strL.charAt(i) == 'x' ||
                    strL.charAt(i) == '/' || strL.charAt(i) == '='){
                rel += strL.charAt(i);
            }
            else rel += hm.get(strL.charAt(i) + "");
        }
 
        return rel;
    }
 
    static Long calcul(long l1, long l2, char op) {
        long rel = 0L;
        switch (op) {
            case '+':
                rel = l1 + l2;
                break;
            case '-':
                rel = l1 - l2;
                break;
            case 'x':
                rel = l1 * l2;
                break;
            case '/':
                rel = l1 / l2;
                break;
        }
        return rel;
    }
 
    static String stringToNum(String target) {
        String rel = "";
        String num = "";
        int s = 0, e = 0;
 
        while (true) {
            e++;
            String mid = target.substring(s, e);
            if (hm.get(mid) != null) {
                rel += mid;
                num += hm.get(mid);
                s = e;
            }
            if(e == target.length()) break;
        }
 
        if(rel.length() == target.length()) return num;
        return null;
 
    }
 
    static void input() throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        s = br.readLine();
 
        for (int i = 0; i < s.length(); i++) {
            if(s.charAt(i) == '+' || s.charAt(i) == '-' ||s.charAt(i) == 'x' ||
                    s.charAt(i) == '/' || s.charAt(i) == '='){
                op.add(s.charAt(i));
            }
        }
 
        hm.put("ZERO", "0");
        hm.put("ONE", "1");
        hm.put("TWO", "2");
        hm.put("THREE", "3");
        hm.put("FOUR", "4");
        hm.put("FIVE", "5");
        hm.put("SIX", "6");
        hm.put("SEVEN", "7");
        hm.put("EIGHT", "8");
        hm.put("NINE", "9");
 
        hm.put("0", "ZERO");
        hm.put("1", "ONE");
        hm.put("2", "TWO");
        hm.put("3", "THREE");
        hm.put("4", "FOUR");
        hm.put("5", "FIVE");
        hm.put("6", "SIX");
        hm.put("7", "SEVEN");
        hm.put("8", "EIGHT");
        hm.put("9", "NINE");
    }
}