문제 설명
코니는 영어 단어가 적힌 카드 뭉치 두 개를 선물로 받았습니다. 코니는 다음과 같은 규칙으로 카드에 적힌 단어들을 사용해 원하는 순서의 단어 배열을 만들 수 있는지 알고 싶습니다.
원하는 카드 뭉치에서 카드를 순서대로 한 장씩 사용합니다.
한 번 사용한 카드는 다시 사용할 수 없습니다.
카드를 사용하지 않고 다음 카드로 넘어갈 수 없습니다.
기존에 주어진 카드 뭉치의 단어 순서는 바꿀 수 없습니다.
예를 들어 첫 번째 카드 뭉치에 순서대로 ["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의 문자열들은 모두 알파벳 소문자로만 이루어져 있습니다.
입출력 예
| cards1 | cards2 | goal | result |
|---|---|---|---|
| ["i", "drink", "water"] | ["want", "to"] | ["i", "want", "to", "drink", "water"] | "Yes" |
| ["i", "water", "drink"] | ["want", "to"] | ["i", "want", "to", "drink", "water"] | "No" |
입출력 예 설명
입출력 예 #1
- 본문과 같습니다.
입출력 예 #2
- cards1에서 "i"를 사용하고 cards2에서 "want"와 "to"를 사용하여 "i want to"까지는 만들 수 있지만 "water"가 "drink"보다 먼저 사용되어야 하기 때문에 해당 문장을 완성시킬 수 없습니다. 따라서 "No"를 반환합니다.
🎯 풀이 과정
- goal 단어 하나씩 가져와서
- cards1 또는 cards2의 첫번째 원소에 있는지 확인
- 있으면 다음 단어 확인
- 없으면 "No" 반환하고 끝내기
def solution1(cards1, cards2, goal):
for s in goal:
if cards1 and s == cards1[0]:
cards1.pop(0)
elif cards2 and s == cards2[0]:
cards2.pop(0)
else:
return "No"
return "Yes"
우왕좌왕 헤맨 썰
처음에 조건문을 제대로 작성하지 않아서 out of range 오류가 떴다 ..
# 조건문을 이렇게만 작성했었음
if s == cards1[0]:
...
elif s == cards2[0]:
...cards1과 cards2 길이가 서로 다른데 이걸 어떻게 체크하고 넘어가야 하는지에 대해 고민하느라 시간이 좀 걸렸다.
cards1, 2 리스트가 비어있나 안비어있나를 체크했으면 됐을 일이다.
그래서 빈 리스트인지 체크할 수 있도록 조건문을 수정해줬다 !
다른 풀이 방법
인덱스를 통해, 단어가 순서대로 사용되었는지 확인하기
- cards 리스트 단어를 하나씩 가져와서
- goal에서 몇 번째 인덱스에 존재하는지 확인
- 단어가 순서대로 사용되었다면, 인덱스가 오름차순으로 정렬되어 있을 것이고
- 그렇지 않다면 정렬되어 있지 않을 것.
def solution3(cards1, cards2, goal):
cards1_index = []
cards2_index = []
for word in cards1:
cards1_index.append(goal.index(word))
for word in cards2:
cards2_index.append(goal.index(word))
if cards1_index == sorted(cards1_index) and cards2_index == sorted(cards2_index):
return "Yes"
else:
return "No"
첫번째 오류.
cards# 리스트에 있는 단어를 모두 사용하지 않을 수 있다.
cards# 리스트 단어를 모두 사용한다는 조건하에 코드를 작성했다.
하지만 위와 같이 런타임 에러 발생.
반례를 작성하여 살펴보니, cards#에 있는 단어가 goal에 없을 수도 있음.
이 부분을 체크하지 않고 넘어가서 런타임 에러가 발생한 것.
첫번째 오류 발견 후 첫번째 수정.
def solution3(cards1, cards2, goal):
cards1_index = []
cards2_index = []
for word in cards1:
if word in goal:
cards1_index.append(goal.index(word))
for word in cards2:
if word in goal:
cards2_index.append(goal.index(word))
if cards1_index == sorted(cards1_index) and cards2_index == sorted(cards2_index):
return "Yes"
else:
return "No"
if문을 추가하여 cards# 단어가 goal에 사용되었는지 확인하고 goal에서 해당 단어의 index를 가져올 수 있게 했다.
그 결과 몇 가지 테스트 케이스를 바로 잡을 수 있었으나, 예외사항이 또 있는 것 같다.
조금 더 고민해봐야 할 방법인 것 같다.
(방법을 찾으면 추가할 예정.)
