설명
현수는 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
문제 설명
첫 번째 줄에 입력받는 과목들이 필수과목이고, 순서대로 선수과목이라고 생각하면 된다.
이 문제는 큐로 접근하면 된다. 큐가 FIFO 구조이므로 필수과목을 큐에 넣어놓고 현수가 짠 수업 설계를 비교하며, 현수가 필수과목을 넣어놨으면 큐에서 제거하면서 다음 순서를 비교해가면 된다
Queue<Character> mustQueue = new LinkedList<>();
Queue<Character> subjectQueue = new LinkedList<>();
for (int i = 0; i < must.length(); i++) {
mustQueue.offer(must.charAt(i));
}
for (int i = 0; i < subject.length(); i++) {
if(!mustQueue.isEmpty() && subject.charAt(i) == mustQueue.peek()) // 필수 과목 큐에 있는 거랑 같아 질때 필수 과목 큐에서 제거하고 다음 순서 비교
mustQueue.poll();
}
순서가 정해져 있으므로, 선수과목을 잘 지켜 시간표를 짰으면, 모든 비교가 끝났을 때 필수 과목의 큐는 비어있을 것이고, 지키지 않았으면 큐는 비어있지 않을 것이다.
if(mustQueue.isEmpty())
System.out.println("YES");
else
System.out.println("NO");
최종 코드
import java.util.LinkedList;
import java.util.Queue;
import java.util.Scanner;
public class P07_교육과정설계 {
public static void main(String[] args) {
//선수 과목 존재
Scanner in = new Scanner(System.in);
String must = in.nextLine();
String subject = in.nextLine();
Queue<Character> mustQueue = new LinkedList<>();
Queue<Character> subjectQueue = new LinkedList<>();
for (int i = 0; i < must.length(); i++) {
mustQueue.offer(must.charAt(i));
}
for (int i = 0; i < subject.length(); i++) {
if(!mustQueue.isEmpty() && subject.charAt(i) == mustQueue.peek()) // 필수 과목 큐에 있는 거랑 같아 질때 필수 과목 큐에서 제거하고 다음 순서 비교
mustQueue.poll();
}
if(mustQueue.isEmpty())
System.out.println("YES");
else
System.out.println("NO");
}
}
'알고리즘 문제풀이 입문: 코딩테스트 대비 > 섹션 5. Stack, Queue(자료구조)' 카테고리의 다른 글
응급실 (1) | 2024.04.15 |
---|---|
공주 구하기 (1) | 2024.04.15 |
쇠막대기 (0) | 2024.04.13 |
후위식 연산(postfix) (0) | 2024.04.12 |
크레인 인형 뽑기 (카카오) (0) | 2024.04.12 |