[프로그래머스 Summer/Winter Coding(~2018) ]영어 끝말잇기

문제

문제링크

풀이

def solution(n, words):
    answer = [0,0]
    for w in range(0,len(words)):
        if (words[w] in words[0:w]) or (w>0 and (words[w][0]!=words[w-1][-1])): 
            w=w+1
            if (w%n)==0:
                answer[0]=(w%n)+n
                answer[1]=int(w/n)
                break
            else:
                answer[0]=(w%n)
                answer[1]=int(w/n)+1
                break

    return answer

오류 발생시 추가 테스트 케이스

• 2,["land", "dream", "mom", "mom", "ror"],[2,2]
• 2,["land", "dream", "mom", "mom"],[2,2]

출처: 프로그래머스