[PS] 99클럽 코테 스터디 11일차 TIL (카드뭉치)

문제

문제 링크

풀이

카드 뭉치들의 순서를 바꿀 수 없고 항상 위에서만 순서대로 뽑아야 한다는 조건이 있어 Queue 로 풀면 될 것 같다. 문제에서 주어지는 세 종류의 카드 뭉치를 모두 각각 Queue 에 넣고 각각의 카드뭉치의 맨 위 카드를 하나씩 꺼내 goal 과 같다면 하나씩 제거하는 방식으로 풀었다.

goalQ 가 모두 비었다면 조건문에 의해 일치하는 카드가 모두 제거된 것이기 때문에 “Yes” 를 출력하고, card1Queue 와 card2Queue 에서 꺼내서 비교하는 과정에서 서로 다른 카드가 발견되는 즉시 “No” 를 리턴한다.

class Solution {

  private static String check(Queue<String> card1Queue, Queue<String> card2Queue, Queue<String> goalQ) {
    while (!goalQ.isEmpty()) {
      if (!card1Queue.isEmpty() && card1Queue.peek().equals(goalQ.peek())) {
        card1Queue.poll();
        goalQ.poll();
      } else if (!card2Queue.isEmpty() && card2Queue.peek().equals(goalQ.peek())) {
        card2Queue.poll();
        goalQ.poll();
      } else {
        return "No";
      }
    }
    return "Yes";
  }

  public String solution(String[] cards1, String[] cards2, String[] goal) {

    Queue<String> card1Queue = new ArrayDeque<>(List.of(cards1));
    Queue<String> card2Queue = new ArrayDeque<>(List.of(cards2));
    Queue<String> goalQ = new ArrayDeque<>(List.of(goal));

    return check(card1Queue, card2Queue, goalQ);
  }
}