코딩 테스트(JAVA)/인프런 문제풀이

교육과정 설계

Lea Hwang 2022. 3. 16. 23:06

설명

현수는 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

 

문제 이해

 

 

수도 코드 및 내가 짠 코드 : Time Limit

import java.util.*;

public class Education {
	public String solution(String n, String m) {
		String answer = "";
		// Q 선언 후 n 넣기
		Queue<Character> Q = new LinkedList<>();
		for(char x : n.toCharArray()) {
			Q.offer(x);
		}
		// m을 배열로 만든 후 하나씩 탐색
		for(char y : m.toCharArray()) {
			while(Q.contains(y)) {
				// m이 Q에 있는지, 있다면 Q의 맨 앞에 있는지 -> poll
				if(y == Q.peek()) Q.poll();
			}
		}
		// m을 다 탐색했을 때 Q가 비어있다면, Yes return
		if(Q.isEmpty()) {
			answer = "YES";
		}else {
			answer = "NO";
		}
		
		return answer;
	}

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

}

 

선생님 정답 코드

import java.util.*;

public class Education {
	public String solution(String n, String m) {
		String answer = "YES";
		// Q 선언 후 n 넣기
		Queue<Character> Q = new LinkedList<>();
		for(char x : n.toCharArray()) {
			Q.offer(x);
		}
		// m을 배열로 만든 후 하나씩 탐색
		for(char y : m.toCharArray()) {
			// 필수과목 체크
			if(Q.contains(y)) {
				// m이 Q의 맨 앞에 있는지 확인
				if(y != Q.poll()) return "NO";
			}
		}
		// m을 다 탐색했을 때 Q가 비어있지 x, NO return
		if(!Q.isEmpty()) {
			answer = "NO";
		}
		return answer;
	}

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

}

 

💡 return을 “YES”로 초기화한 후, “NO”의 경우만 적어준다.

if문 안에 if문 넣는거 괜찮음

if(y != Q.poll()) return "NO"; 어짜피 꺼낼 거니까 꺼내서 확인

 

'코딩 테스트(JAVA) > 인프런 문제풀이' 카테고리의 다른 글

좌표 정렬  (0) 2022.03.24
응급실  (0) 2022.03.16
공주 구하기  (0) 2022.03.16
쇠막대기  (0) 2022.03.16
후위식 연산(postfix)  (0) 2022.03.16