영어 끝말잇기

문제 

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

프로그래머스

 

 

풀이 

"""
문제해결방법
1. 중복을 검사하는 리스트 한개 생성
2. 몇번째 사람인지 -> for문 i % n + 1
3. 몇번째 차례인지 -> for문 i // n + 1
3. 앞단어의 마지막 글자가 그 다음단어의 첫글짜와 동일한지 여부 확인, 중복여부 확인
"""
def solution(n, words):
    answer = []
    same_thing = []
    
    for i in range(len(words)):
    	# 처음 단어는 그냥 추가
        if i == 0:
            same_thing.append(words[i])
        else:
        	# 중복여부, 앞글자랑 뒤글자 일치 여부 확인
            if (words[i] in same_thing or 
                same_thing[-1][-1] != words[i][0]):
                human_idx = i % n + 1
                count = i // n + 1
                answer.append(human_idx)
                answer.append(count)
                break
            # 문제가 없으면 그냥 추가
            else:
                same_thing.append(words[i])
                
    # 문제없이 끝말잇기가 끝났을 경우
    if len(answer) == 0:
        for _ in range(2):
            answer.append(0)
            
    return answer