코딩 테스트/Java

[JAVA]7. 교육과정 설계 (Queue)

반응형

설명

현수는 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“를 출력합니다.

 

예시 입출력

CBA
CBDAGE
YES

 

설명

queue의 구조(FIFO)만 알면 쉽게 풀 수 있는 문제이다.

필수과목을 queue에 순서대로 넣어준다.
수업계획을 순차적으로 탐색하면서 queue제일 맨 위에 있는 값과 확인하여, 값이 같다면 queue에서 제거
queue에 값이 없으면 조건 성립 return "YES"

import java.util.LinkedList;
import java.util.Queue;
import java.util.Scanner;

public class Main{
    public static String solution(String need, String plan){
        String answer = "NO";
        Queue<Character> q = new LinkedList<>();
        for(char x : need.toCharArray()) q.offer(x);
        for(char x : plan.toCharArray()){
            if(q.peek() == x) q.poll();
            if(q.isEmpty()) return "YES";
        }
        return answer;
    }

    public static void main(String[] args) {
        Scanner scan = new Scanner(System.in);
        String need = scan.next();
        String plan = scan.next();
        System.out.println(solution(need, plan));
    }
}
/*
다른풀이
public static 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)){
                if(x!=q.poll()) return "NO";
            }
        }
        if(!q.isEmpty()) return "NO";
        return answer;
    }
 */
반응형