[Java] BOJ 17479 정식당

17479번: 정식당

---

정답 코드

• 문제에서 요구하는대로 구현
• Map을 활용해서 가격 정보 저장
• 문제에서 메뉴판에서 주어진 메뉴만 주문하기 때문에 서비스 메뉴는 별도 저장 필요 X

import java.io.*;
import java.util.*;

public class Main {

    static BufferedReader br;
    static BufferedWriter bw;
    static StringTokenizer st;
    static int A, B, C, N;
    static HashMap<String, Integer> nMenu, sMenu;

    public static void main(String[] args) throws IOException {
        br = new BufferedReader(new InputStreamReader(System.in));
        bw = new BufferedWriter(new OutputStreamWriter(System.out));

        st = new StringTokenizer(br.readLine(), " ");
        A = Integer.parseInt(st.nextToken());
        B = Integer.parseInt(st.nextToken());
        C = Integer.parseInt(st.nextToken());

        nMenu = new HashMap<>();
        for (int i = 0; i < A; i++) {
            st = new StringTokenizer(br.readLine(), " ");
            nMenu.put(st.nextToken(), Integer.parseInt(st.nextToken()));
        }

        sMenu = new HashMap<>();
        for (int i = 0; i < B; i++) {
            st = new StringTokenizer(br.readLine(), " ");
            sMenu.put(st.nextToken(), Integer.parseInt(st.nextToken()));
        }

        for (int i = 0; i < C; i++) {
            String name = br.readLine();
        }

        N = Integer.parseInt(br.readLine());
        long a = 0, b = 0, c = 0;
        for (int i = 0; i < N; i++) {
            String order = br.readLine();
            if (nMenu.containsKey(order)) {
                a += nMenu.get(order);
            } else if (sMenu.containsKey(order)) {
                b += sMenu.get(order);
            } else {
                c++;
            }
        }

        if (c > 1 || (b > 0 && a < 20000) || (c > 0 && a + b < 50000)) {
            System.out.println("No");
        } else {
            System.out.println("Okay");
        }
    }
}