[프로그래머스] 카드 뭉치

문제 설명

코니는 영어 단어가 적힌 카드 뭉치 두 개를 선물로 받았습니다. 코니는 다음과 같은 규칙으로 카드에 적힌 단어들을 사용해 원하는 순서의 단어 배열을 만들 수 있는지 알고 싶습니다.

원하는 카드 뭉치에서 카드를 순서대로 한 장씩 사용합니다.
한 번 사용한 카드는 다시 사용할 수 없습니다.
카드를 사용하지 않고 다음 카드로 넘어갈 수 없습니다.
기존에 주어진 카드 뭉치의 단어 순서는 바꿀 수 없습니다.
예를 들어 첫 번째 카드 뭉치에 순서대로 ["i", "drink", "water"], 두 번째 카드 뭉치에 순서대로 ["want", "to"]가 적혀있을 때 ["i", "want", "to", "drink", "water"] 순서의 단어 배열을 만들려고 한다면 첫 번째 카드 뭉치에서 "i"를 사용한 후 두 번째 카드 뭉치에서 "want"와 "to"를 사용하고 첫 번째 카드뭉치에 "drink"와 "water"를 차례대로 사용하면 원하는 순서의 단어 배열을 만들 수 있습니다.

문자열로 이루어진 배열 cards1, cards2와 원하는 단어 배열 goal이 매개변수로 주어질 때, cards1과 cards2에 적힌 단어들로 goal를 만들 있다면 "Yes"를, 만들 수 없다면 "No"를 return하는 solution 함수를 완성해주세요.

제한사항

• 1 ≤ cards1의 길이, cards2의 길이 ≤ 10

  • 1 ≤ cards1[i]의 길이, cards2[i]의 길이 ≤ 10

  • cards1과 cards2에는 서로 다른 단어만 존재합니다.

• 2 ≤ goal의 길이 ≤ cards1의 길이 + cards2의 길이

  • 1 ≤ goal[i]의 길이 ≤ 10
  • goal의 원소는 cards1과 cards2의 원소들로만 이루어져 있습니다.
    cards1, cards2, goal의 문자열들은 모두 알파벳 소문자로만 이루어져 있습니다.

입출력 예

입출력 예 설명

입출력 예 #1

본문과 같습니다.

입출력 예 #2

cards1에서 "i"를 사용하고 cards2에서 "want"와 "to"를 사용하여 "i want to"까지는 만들 수 있지만 "water"가 "drink"보다 먼저 사용되어야 하기 때문에 해당 문장을 완성시킬 수 없습니다. 따라서 "No"를 반환합니다.

정답코드

function solution(cards1, cards2, goal) {
    let idx1 = 0;
    let idx2 = 0;
    
    let count = 0;
    let n = goal.length;
    for (let i = 0; i < goal.length; i++) {
        if (idx1 < cards1.length && goal[i] === cards1[idx1]) {
            count++;
            idx1++; // cards1에서 카드를 사용한 경우
        } else if (idx2 < cards2.length && goal[i] === cards2[idx2]) {
            count++;
            idx2++; // cards2에서 카드를 사용한 경우
        } 
    }

    return count === n ? "Yes": "No"; // 모든 goal 요소를 성공적으로 찾은 경우
}

코드설명

원하는 단어 배열 goal의 길이만큼 반복문을 실행하면서 각 카드뭉치 시작점에 존재하면 count++ 한 뒤 마지막에 goal의 문장을 만들었다면 count가 goal 배열 길이만큼 증가했을 것이다. cards1과 cards2 배열의 길이가 다르므로 각각 인덱스 시작점인 0을 변수로 두고 count++ 할때 해당 idx 값도 증가시켰다.

다른 사람의 풀이

function solution(cards1, cards2, goal) {
    let j = 0;
    let k = 0;
    for(let i=0;i<goal.length;i++){
        if(goal[i] == cards1[j]) j++;
        else if(goal[i] == cards2[k]) k++;
        else return "No"
    }
    return "Yes";
}

내 정답 코드와 비교했을 때 더 간단하고 가독성도 좋은 것 같다. 여기서는 각 카드의 index 시작점은 똑같이 설정해 주지만 인덱스의 범위를 확인하지 않는데 왜 그러냐면 만약 만들 수 있는 goal 문장이 매개변수로 주어지면 index 범위를 초과할 상황 없이 반복문에서 return "No" 하지 않고 정상적으로 종료될 것이다. 그리고 반복문에 else return "No"를 해줌으로써 일치하는 단어가 없다면 이후에 반복을 할 필요 없이 조기 종료도 가능하다.

다른 사람의 풀이를 보는 것만으로도 많은 도움이 되는 것 같다. 더 효율적인 방법을 알게되는 것은 물론 나중에 다른 문제를 풀 때도 응용도 가능하다.