[programmers] 뉴스 클러스터링 - Counter

https://programmers.co.kr/learn/courses/30/lessons/17677#

코딩테스트 연습 - [1차] 뉴스 클러스터링

 

 

 

첫 코드

from collections import Counter

def solution(str1, str2):
    string1, string2 = [] ,[]
    
    string = ['a','b','c','d','e','f','g','h','i','j','k','l','m',
             'n','o','p','q','r','s','t','u','v','w','x','y','z']
    
    for i in range(len(str1)-1):
        if str1[i].lower() in string and str1[i+1].lower() in string:
            string1.append(str1[i].lower()+str1[i+1].lower())

    for i in range(len(str2)-1):
        if str2[i].lower() in string and str2[i+1].lower() in string:
            string2.append(str2[i].lower()+str2[i+1].lower())

    
    inter = set(string1)&set(string2)
    untion = set(string2+string1)
    
    if len(inter) == 0 and len(union) == 0:
        return 1*65536
    else:
        return int((len(inter)/len(union))*65536)

집합으로는 문제에 나와 있는 다중집합 허용 조건을 만족하지 못한다.

코드를 이리저리 수정해보아도 테스트 케이스는 통과하지만 제출 후 실패하는 케이스들이 많다.

중복되는 요소에 대해서는 고유번호를 추가적으로 더 붙이는 방법도 생각했지만 이것도 실패..

 

 

 

수정 후 코드 

집합 set 이 아닌 Counter 사용

from collections import Counter

def solution(str1, str2):
    string1, string2 = [] ,[]
    
    string = ['a','b','c','d','e','f','g','h','i','j','k','l','m',
             'n','o','p','q','r','s','t','u','v','w','x','y','z']
    
    for i in range(len(str1)-1):
        if str1[i].lower() in string and str1[i+1].lower() in string:
            string1.append(str1[i].lower()+str1[i+1].lower())

    for i in range(len(str2)-1):
        if str2[i].lower() in string and str2[i+1].lower() in string:
            string2.append(str2[i].lower()+str2[i+1].lower())

    
    inter = list((Counter(string1) & Counter(string2)).elements())
    union = list((Counter(string1) | Counter(string2)).elements())
    
    if len(inter) == 0 and len(union) == 0:
        return 1*65536
    else:
        return int((len(inter)/len(union))*65536)

 

Counter 에 사용한 건 이번이 처음!

https://dongdongfather.tistory.com/70

[파이썬 기초] Counter를 이용한 항목 계산

의 글을 참고했다.