영아일지

[프로그래머스] level 2 - 1차 뉴스 클러스터링 본문

디지털/코딩테스트

[프로그래머스] level 2 - 1차 뉴스 클러스터링

영아일지 2023. 4. 2. 01:03

2018 KAKAO Blind recruit [1차 뉴스 클러스터링]

 

sol)

1. 대소문자를 구분안하기 때문에 모두 대문자로 변환한다.

2. 알파벳이외에는 다중집합을 만들지 않기 때문에 알파벳이라면 다중집합인 s1, s2 를 만들었다

3. s1과 s2가 공집합이면 자카드 유사도가 1 이라 미리 가지치기 했다.

4. 교집합일 경우를 구해주었다. 

   - 이때, break를 해주어야 중복이 되지 않으며 교집합 값을 remove해주어야 한다.

5. s2는 이미 교집합이 제거된 상태이기 때문에 합집합은 s1길이 + s2길이이다.

6. math를 활용하여 버림한다.

import math
def solution(str1, str2):
    answer = 0
    inter = 0
    str1 = str1.upper() #대소문자 구분안함
    str2 = str2.upper()
    
    #다중집합
    s1 = [str1[i1:i1+2] for i1 in range(len(str1)-1) if str1[i1:i1+2].isalpha()]
    s2 = [str2[i2:i2+2] for i2 in range(len(str2)-1) if str2[i2:i2+2].isalpha()]
    
    #공집합이면 1
    if s1 == [] and s2 == []:
        return 65536
    
    for one in range(len(s1)):
        for two in range(len(s2)):
            if s1[one] == s2[two]:
                inter += 1
                s2.remove(s2[two])
                break
            
    #자카드 유사도
    uni = len(s1)+len(s2)
    
    answer = inter/uni
    return math.trunc(answer*65536)

느낀 점)

여러가지 함수를 활용을 한 풀이다. 문제풀이를 적으면서 리팩토링을 해보았다. 

군더더기없는 코드를 짜고 싶다!

 

다른 사람 풀이)

 

import re
import math

def solution(str1, str2):
    str1 = [str1[i:i+2].lower() for i in range(0, len(str1)-1) if not re.findall('[^a-zA-Z]+', str1[i:i+2])]
    str2 = [str2[i:i+2].lower() for i in range(0, len(str2)-1) if not re.findall('[^a-zA-Z]+', str2[i:i+2])]

    gyo = set(str1) & set(str2)
    hap = set(str1) | set(str2)

    if len(hap) == 0 :
        return 65536

    gyo_sum = sum([min(str1.count(gg), str2.count(gg)) for gg in gyo])
    hap_sum = sum([max(str1.count(hh), str2.count(hh)) for hh in hap])

    return math.floor((gyo_sum/hap_sum)*65536)