2022. 3. 16. 23:06ㆍ코딩 테스트(JAVA)/인프런 문제풀이
설명
현수는 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 |