[프로그램스] 영어 끝말잇기

나의 코드

def solution(n, words):
    answer = [words[0]]
    x, y = 1, 1
    for i in range(1, len(words)):
        x += 1
        if x == n + 1:
            x, y = 1, y + 1

        if words[i] in answer:
            return [x, y]
        elif answer[-1][-1] != words[i][0]:
            return [x, y]
        else:
            answer.append(words[i])

    return [0, 0]

결과 인덱스를 x, y로 따로 연산해줬는데 그냥 나눠서 바로 구할 수 있었다.

 

 

효율 코드

return [(p%n)+1, (p//n)+1]