[Python] 프로그래머스. 영어 끝말잇기 (Level 2)

https://programmers.co.kr/learn/courses/30/lessons/12981

코딩테스트 연습 - 영어 끝말잇기 | 프로그래머스

 

주어진 조건대로 구현하면 되는 문제.

 

 

 

 

	def solution(n, words):
	# 처음으로 말한 단어 저장
	dataset = set([words[0]])
	# 이전 단어 기억
	prev_words = words[0]
	for i in range(1,len(words)):
	# 이전 단어의 마지막 글자와 제시단어의 첫글자가 다르거나, 이미 있는 단어일 경우
	if (prev_words[-1] != words[i][0]) or words[i] in dataset:
	# 사람, 차례
	return [(i % n)+1, (i//n)+1]
	# 등장한 단어 저장
	dataset.add(words[i])
	# 제시단어 기억
	prev_words = words[i]
	return [0,0]

view raw [프로그래머스] 영어 끝말잇기.py hosted with ❤ by GitHub