[level 1] 카드 뭉치

문제

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

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

예를 들어 첫 번째 카드 뭉치에 순서대로 ["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

1) len(goal)의 길이 만큼 for문을 반복한다.

2) cards1의 p1번째 원소나 cards2의 p2번째 원소가 goal의 i번째 원소와 같은지 확인한다. 만약 같다면 해당 원소의 p1의 값을 올려 다른 원소를 탐색할 수 있게 한다.

3) 각 원소가 같다면 sentence에 +1을 한다. 

4) 만약 len(cards1)이나 len(cards2)의 길이보다 p1이나 p2의 길이가 더 크다면 0으로 초기화 해준다(딱히 의미 없음...원래 break했는데 그러니까 옳은 케이스에서도 제대로 못돌고 나와서 그런거임..)

5) len(goal)과 setence가 같은지 확인한다. 같다면 Yes를, 다르다면 No를 반환한다. 

def solution(cards1, cards2, goal):
    answer = ''
    sentence = 0

    p1, p2 = 0, 0

    for i in range(len(goal)):
    
        if cards1[p1] == goal[i]:
            sentence +=1
            p1 += 1
            if p1 > len(cards1)-1: p1 = 0
    
        elif cards2[p2] == goal[i]:
            sentence +=1
            p2+=1
            if p2 > len(cards2)-1: p2 =0 
            
    if len(goal) == sentence:
        return "Yes"
    else:
        return "No"

 

---

다른 분들 풀이

def solution(cards1, cards2, goal):
    for g in goal:
        if len(cards1) > 0 and g == cards1[0]:
            cards1.pop(0)       
        elif len(cards2) >0 and g == cards2[0]:
            cards2.pop(0)
        else:
            return "No"
    return "Yes"

오 아예 시작부터 길이를 확인하고 pop을 사용해서 옳으면 pop으로 빼는 거 좋은 것 같다.

그리고 아예 다 같지 않으면 바로 No를 return하는 것도!

이런 문제 나오면 꼭 pop 사용해서 써보기로...