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

문제

카드 뭉치 : 문제 링크

문제 분석

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

• 원하는 카드 뭉치에서 카드를 순서대로 한 장씩 사용한다.
• 한 번 사용한 카드는 다시 사용할 수 없다.
• 카드를 사용하지 않고 다음 카드로 넘어갈 수 없다.
• 기존에 주어진 카드 뭉치의 단어 순서는 바꿀 수 없다.

• 예를 들어 첫 번째 카드 뭉치에 순서대로 ["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의 문자열들은 모두 알파벳 소문자로만 이루어져 있다.

• goal 내에 cards1, cargs2의 원소가 존재하는지 확인하기 위해 find() 함수를 사용하므로 algorithm 헤더를 include
• cards1과 cards2의 이전 인덱스를 저장할 정수형 변수 c1, c2를 각각 -1로 초기화. for loop를 통해 goal의 첫번째부터 마지막 원소까지 순환하고, if문을 사용하여 현재 인덱스의 원소를 cards1에서 찾았을 경우 정수형 변수 c1_pos에 찾은 인덱스를 저장. 이후 if문을 사용하여 c1이 c1_pos보다 크거나 c1_pos가 c1보다 1만큼 크지 않다면(카드를 뛰어넘었음을 의미하므로) "No"를 return. 이외의 경우는 c1에 c1_pos값을 저장. else if문을 사용하여 현재 인덱스의 원소를 cards2에서 찾았을 경우에도 마찬가지로 구성. loop를 탈출 했다면, 모든 조건을 만족했음을 의미하므로 "Yes"를 return

1. vector내에 해당 원소가 존재하는지 확인
=> find(v.begin(), v.end(), 찾을 대상)
1) return 값이 v.end()인 경우 => 해당 원소가 존재하지 않음
2) return 값이 v.end()가 아닌 경우 => 해당 원소가 존재함
2. vector내에서 해당 원소가 위치하는 인덱스 찾기
=> find(v.begin(), v.end(), 찾을 대상) - v.begin

풀이

#include <string>
#include <vector>
#include <algorithm>

using namespace std;

string solution(vector<string> cards1, vector<string> cards2, vector<string> goal) {
    int c1 = -1, c2 = -1;
    
    for(int i = 0; i < goal.size(); ++i) {
        if(find(cards1.begin(), cards1.end(), goal[i]) != cards1.end()) {
            int c1_pos = find(cards1.begin(), cards1.end(), goal[i]) - cards1.begin();
            if(c1 > c1_pos || c1_pos - 1 != c1) return "No";
            else c1 = c1_pos;
        }
        else if(find(cards2.begin(), cards2.end(), goal[i]) != cards2.end()) {
            int c2_pos = find(cards2.begin(), cards2.end(), goal[i]) - cards2.begin();
            if(c2 > c2_pos || c2_pos - 1 != c2) return "No";
            else c2 = c2_pos;
        }
    }
    return "Yes";
}