[C++, python] 프로그래머스 - 카드 뭉치

문제 링크

프로그래머스

문제 풀이

이 문제는 cards1과 cards2의 순서에 맞게 goal이 나오는지 확인하면 된다. 순서가 뒤바뀐 경우, "No"를 반환한다.

정답 코드

C++

#include <string>
#include <vector>

using namespace std;

string solution(vector<string> cards1, vector<string> cards2, vector<string> goal) {
    int i = 0, j = 0;
    while (i + j < goal.size()) {
        if (i < cards1.size() && cards1[i] == goal[i + j])
            i++;
        else if (j < cards2.size() && cards2[j] == goal[i + j])
            j++;
        else return "No";
    }
    return "Yes";
}

python

def solution(cards1, cards2, goal):
    i = 0
    j = 0
    while i + j < len(goal) :
        if i < len(cards1) and cards1[i] == goal[i + j] :
            i += 1
        elif j < len(cards2) and cards2[j] == goal[i + j] :
            j += 1
        else :
            return "No"
    return "Yes"