[프로그래머스] Lv2. 뉴스 클러스터링

문제 링크: https://school.programmers.co.kr/learn/courses/30/lessons/17677

프로그래머스

✍🏻코드

from collections import Counter

def solution(str1, str2):
    
    arr1,arr2 = [],[]
    for i in range(len(str1)-1):
        if str1[i:i+2].isalpha():
            arr1.append(str1[i:i+2].upper())
    for i in range(len(str2)-1):
        if str2[i:i+2].isalpha():
            arr2.append(str2[i:i+2].upper())
  
    if len(arr1) == 0 and len(arr2) ==0 :
        return 65536
    
    c1 = Counter(arr1)
    c2 = Counter(arr2)
    
    inter = sum((c1&c2).values())
    union = sum((c1|c2).values())
    
    return int(inter/union*65536)

💡풀이

• Counter의 사용법을 알면 쉽게 풀 수 있다.
• Counter(arr1)을 하게되면 arr1에 있는 데이터가 각각 몇개 있는지 딕셔너리의 형태로 반환해준다.
• c1 & c2 를 하게 되면 교집합을 알 수 있고, 우리가 필요한 것은 교집합의 개수의 합이므로 values()의 sum()을 이용해 교집합의 개수의 합을 구해준다.
• 마찬가지로 c1|c2로 합집합을 구할수 있고, sum((c1|c2.values())로 합집합의 개수의 합을 구할 수 있다.