문제
코니는 영어 단어가 적힌 카드 뭉치 두 개를 선물로 받았습니다. 코니는 다음과 같은 규칙으로 카드에 적힌 단어들을 사용해 원하는 순서의 단어 배열을 만들 수 있는지 알고 싶습니다.
원하는 카드 뭉치에서 카드를 순서대로 한 장씩 사용합니다.
한 번 사용한 카드는 다시 사용할 수 없습니다.
카드를 사용하지 않고 다음 카드로 넘어갈 수 없습니다.
기존에 주어진 카드 뭉치의 단어 순서는 바꿀 수 없습니다.
예를 들어 첫 번째 카드 뭉치에 순서대로 ["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 함수를 완성해주세요.
https://school.programmers.co.kr/learn/courses/30/lessons/159994
접근
1. goal[0] 과 cards1[0] 을 비교한다.
1-1.같으면 둘 다 pop
1-2.다르다면 goal[0] 과 cards1[1] 을 비교한다.
1-3.같으면 역시 둘 다 pop
1-4 다르다면 No 반환
2. 넘어갔다면 goal[0]로 반복 (삭제했으므로)
또한 list로 pop(0) 하면 리스트 길이만큼 시간이 걸리므로
모든 리스트를 deque로 변환한후 popleft 하는 것이 빠를 듯
해결
def solution(cards1, cards2, goal):
answer = ''
cards1_deq = deque(cards1)
cards2_deq = deque(cards2)
while answer!='No' and goal:
word = goal[0]
if cards1_deq != deque([]) and cards1_deq[0] == word:
cards1_deq.popleft()
goal.remove(word)
elif cards2_deq != deque([]) and cards2_deq[0] == word:
cards2_deq.popleft()
goal.remove(word)
else:
answer = 'No'
if goal == []:
answer = 'Yes'
return answer
다른 사람의 풀이
다른 사람들의 풀이를 보니 상당히 간단해서 내가 너무 어렵게 생각했나, 싶었다.
이럴 때 마다 살짝 자괴감도 들고,, 나는 바보인가 싶기도 하고.. 그렇다.
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"
GitHub 댓글