Python/알고리즘

[파이썬 알고리즘 인터뷰] 4. 가장 흔한 단어

AI 꿈나무 2021. 2. 9. 14:53
반응형

 

 이 포스팅은 파이썬 알고리즘 인터뷰를 공부하면서 정리한 내용입니다.

 

 출처 : 파이썬 알고리즘 인터뷰

 


4. 가장 흔한 단어

리트코드 819번 문제입니다.

leetcode.com/problems/most-common-word/

 

 

풀이

 paragraph를 전처리 하는 과정이 핵심이었습니다.

 banned에 해당하는 문자와 특수기호, 띄어쓰기를 제거해야 합니다.

 파이썬 리스트컴프리헨션을 이용해 한줄로 전처리하는 코드가 인상깊었습니다.

 

def mostCommonWord(self, paragraph, banned):
    words = [word for word in re.sub(r'[^\w]',' ',paragraph)
        .lower().split() if word not in banned]

    counts = collections.Counter(words)
    return counts.most_common(1)[0][0]
반응형