교육과정 정하기

23 Apr 2023 in Algorithm on Java, Coding, Test

문제

설명

현수는 1년 과정의 수업계획을 짜야 합니다.

수업중에는 필수과목이 있습니다. 이 필수과목은 반드시 이수해야 하며, 그 순서도 정해져 있습니다.

만약 총 과목이 A, B, C, D, E, F, G가 있고, 여기서 필수과목이 CBA로 주어지면 필수과목은 C, B, A과목이며 이 순서대로 꼭 수업계획을 짜야 합니다.

여기서 순서란 B과목은 C과목을 이수한 후에 들어야 하고, A과목은 C와 B를 이수한 후에 들어야 한다는 것입니다.

현수가 C, B, D, A, G, E로 수업계획을 짜면 제대로 된 설계이지만

C, G, E, A, D, B 순서로 짰다면 잘 못 설계된 수업계획이 됩니다.

수업계획은 그 순서대로 앞에 수업이 이수되면 다음 수업을 시작하다는 것으로 해석합니다.

수업계획서상의 각 과목은 무조건 이수된다고 가정합니다.

필수과목순서가 주어지면 현수가 짠 N개의 수업설계가 잘된 것이면 “YES”, 잘못된 것이면 ”NO“를 출력하는 프로그램을 작성하세요.

입력

첫 줄에 한 줄에 필수과목의 순서가 주어집니다. 모든 과목은 영문 대문자입니다.

두 번 째 줄부터 현수가 짠 수업설계가 주어집니다.(수업설계의 길이는 30이하이다)

출력

첫 줄에 수업설계가 잘된 것이면 “YES”, 잘못된 것이면 ”NO“를 출력합니다.

예시 입력 1

CBA
CBDAGE

예시 출력 1

YES

해결방법

• 순서가 정해진 과목이수 정보가 있고, 과목 리스트에서 이수 정보에 맞게 순서대로 들었는지 체크 하는 로직을 구현한다.

코드

import java.util.*;

class Main {
    public String solution(String need, String plan) {
        // 맞다고 가정하고 시작.
        String answer = "YES";
        Queue<Character> Q = new LinkedList<>();
        // 큐에 필요정보를 저장한다.
        for (char x : need.toCharArray()) Q.offer(x);
        for (char x : plan.toCharArray()) {
            // 과목 정보 만큼 반복한다.
            // 큐에 해당 과목이 있는지 체크 한다.
            if (Q.contains(x)) {
                // 있다면, 순서에 맞게 poll 했는데 다른 과목이면 틀린 것이므로 NO를 리턴한다.
                if (x != Q.poll()) return "NO";
            }
        }
        // 큐가 남아있어도 NO 를 리턴한다.
        if (!Q.isEmpty()) return "NO";
        return answer;
    }

    public static void main(String[] args) {
        Main.Main T = new Main.Main();
        Scanner kb = new Scanner(System.in);
        String a = kb.next();
        String b = kb.next();
        System.out.println(T.solution(a, b));
    }
}