[Python/String-Manipulation] Minimum Distance Between Two Words 두 단어사이 최소거리 찾기.

문제

단어 리스트와 그 뒤에 이어지는 두 개의 단어가 주어졌을 때, 주어진 두 단어 사이의 최소 거리를 해당 단어 리스트에서 찾으세요.

입출력 예시)

Input: S = [ “Apple”, “Banana”, “Apple”, “Cherry”, “Grape”]

word1 = “Apple”

word2 = “Grape”

Output: 2

 

코드

S = ["Apple", "Grape", "Apple", "Cherry", "Grape"] 
word1 = "Apple"
word2 = "Grape"
distace=[]
count=0

for item in S:
    count+=1
    if item ==word1:
        count=0
    elif item ==word2:
        distace.append(count)

distace.sort()
        

print(distace.pop(0))