문제 설명
1부터 n까지 번호가 붙어있는 n명의 사람이 영어 끝말잇기를 하고 있습니다. 영어 끝말잇기는 다음과 같은 규칙으로 진행됩니다.
- 1번부터 번호 순서대로 한 사람씩 차례대로 단어를 말합니다.
- 마지막 사람이 단어를 말한 다음에는 다시 1번부터 시작합니다.
- 앞사람이 말한 단어의 마지막 문자로 시작하는 단어를 말해야 합니다.
- 이전에 등장했던 단어는 사용할 수 없습니다.
- 한 글자인 단어는 인정되지 않습니다.
다음은 3명이 끝말잇기를 하는 상황을 나타냅니다.
tank → kick → know → wheel → land → dream → mother → robot → tank
위 끝말잇기는 다음과 같이 진행됩니다.
- 1번 사람이 자신의 첫 번째 차례에 tank를 말합니다.
- 2번 사람이 자신의 첫 번째 차례에 kick을 말합니다.
- 3번 사람이 자신의 첫 번째 차례에 know를 말합니다.
- 1번 사람이 자신의 두 번째 차례에 wheel을 말합니다.
- (계속 진행)
끝말잇기를 계속 진행해 나가다 보면, 3번 사람이 자신의 세 번째 차례에 말한 tank 라는 단어는 이전에 등장했던 단어이므로 탈락하게 됩니다.
사람의 수 n과 사람들이 순서대로 말한 단어 words 가 매개변수로 주어질 때, 가장 먼저 탈락하는 사람의 번호와 그 사람이 자신의 몇 번째 차례에 탈락하는지를 구해서 return 하도록 solution 함수를 완성해주세요.
제한사항
- 끝말잇기에 참여하는 사람의 수 n은 2 이상 10 이하의 자연수입니다.
- words는 끝말잇기에 사용한 단어들이 순서대로 들어있는 배열이며, 길이는 n 이상 100 이하입니다.
- 단어의 길이는 2 이상 50 이하입니다.
- 모든 단어는 알파벳 소문자로만 이루어져 있습니다.
- 끝말잇기에 사용되는 단어의 뜻(의미)은 신경 쓰지 않으셔도 됩니다.
- 정답은 [ 번호, 차례 ] 형태로 return 해주세요.
- 만약 주어진 단어들로 탈락자가 생기지 않는다면, [0, 0]을 return 해주세요.
입출력 예
| n | words | result |
|---|---|---|
| 3 | [tank, kick, know, wheel, land, dream, mother, robot, tank] | [3,3] |
| 5 | [hello, observe, effect, take, either, recognize, encourage, ensure, establish, hang, gather, refer, reference, estimate, executive] | [0,0] |
| 2 | [hello, one, even, never, now, world, draw] | [1,3] |
풀이
def solution(n, words):
new_list = []
idx = 0
for i, v in enumerate(words):
if v in new_list:
idx = i + 1
break
if i > 0 and v[0] != words[i-1][-1]:
idx = i + 1
break
else:
new_list.append(v)
return [idx % n if idx % n else n, idx//n+1 if idx % n else idx//n] if idx else[0, 0]-
이번 <끝말잇기>문제의 요건사항을 정리해보면 이렇습니다. 첫째. 같은 문자가 나오면 터집니다. 둘째. 이전 문자열의 마지막 글자와 현재 문자열의 첫 번째 글자가 일치하지 않으면 터집니다. 터질 때의 인덱스를 구해야 합니다. 모든 문자열을 순회해도 터지지 않았다면 [0, 0]을 반환합니다.
-
요건들을 생각하며, 중복검사를 위한 new_list와 터질 때 인덱스를 저장할 idx를 선언합니다.
중복이 되는 경우와 틀린 글자가 나올때 인덱스를 조건문으로 처리해 변수 idx에 저장합니다. 이때 idx는 0부터 시작하므로 실제 현실의 셈법에 맞춰 +1을 해줍니다.(컴퓨터의 첫 번째는 1이 아니라 0이므로)
그리고 반드시 break로 for문을 끝내줘야 합니다. 그렇지 않다면 mother, mother 같이 두 개의 조건 모두를 충족하는 경우 반환되는 idx가 예측값과 달라질 수 있습니다. -
idx를 n으로 나눕니다. 나머지 = 걸린 사람의 번호입니다. 만약 나머지가 없다면(0이라면) n번째 사람이라는 이야기 입니다. 삼항연산자를 활용해 이 경우의 수를 처리해줍니다.
-
idx를 n으로 나눈 몫으로 걸린 번호의 사람이 몇 번째 순번에서 걸렸는지 알 수 있습니다. 이번에도 삼항연산자를 활용합니다. n번째 사람은 무조건 순번상 마지막 사람이므로 몫을 그대로 적용하면 되고, 이외의 경우에는 n번째까지 돌고 나머지 만큼 전체 순번이 '한 번' 더 돌았다는 의미이므로 +1을 해줘야 합니다.
-
그리고 전체 리턴 항에 삼항연산자를 활용해 [0, 0]인 경우의 수도 처리해줍니다.
다른 풀이
def solution(n, words):
for p in range(1, len(words)):
if words[p][0] != words[p-1][-1] or words[p] in words[:p]:
return [(p%n)+1, (p//n)+1]
else:
return [0,0]제가 두 개의 조건문으로 처리한 것을 or을 활용해 더욱 간단하게 적용했네요.
저는 현재의 셈법에 맞춰 조정하고, 조정한 것을 삼항연산자로 삽질을 했는데 여기는 오히려 인덱스를 그대로 써서 더 간단히 풉니다. 이게 가능한 이유는 현실셈법에서 첫 번째 테스트 케이스에서 나머지가 1,2,0(이게 3번째여야함) 0, 1, 2로 변환되기 때문에 나머지에 무조건 +1을 하면 무조건 성립이 되고 몫도 마찬가지로 무조건 성립이 되기 때문입니다. 😱
