[프로그래머스/python] 코딩테스트 연습 Lv.2 영어 끝말잇기

문제

(단순 영어 끝말잇기 설명이라 생략)

 

입출력 예

Solution

def search(words, w):
    for i in words:
        if i==w:
            return 1
    return -1

def solution(n, words):

    for i in range(1, len(words)):
        if words[i-1][-1] != words[i][0] or search(words[:i], words[i]) == 1:
            return [i%n+1, int(i/n)+1]

    return [0, 0]

 

search 함수: 앞 사람이 말한 단어와 같은 단어를 말한 경우를 찾는 함수

- for문을 이용하여 현재 탐색하는 단어의 위치를 제외한 words 내의 단어들과 탐색하고자 하는 단어를 비교했을 때 같은 단어가 있으면 1을 반환

- 없을 시 -1을 반환

 

solution 함수: main

- if문 조건 1: 앞 사람이 말한 단어의 끝과 현재 말한 단어의 시작이 같지 않은 경우

- if문 조건 2: search 함수에서 1을 반환한 경우