[프로그래머스] [1차] 뉴스 클러스터링 / Level 2 / Java

알재·2024년 8월 5일
0

코딩 테스트

목록 보기
52/57
post-custom-banner

링크

문제링크

문제

뉴스 클러스터링

여러 언론사에서 쏟아지는 뉴스, 특히 속보성 뉴스를 보면 비슷비슷한 제목의 기사가 많아 정작 필요한 기사를 찾기가 어렵다. Daum 뉴스의 개발 업무를 맡게 된 신입사원 튜브는 사용자들이 편리하게 다양한 뉴스를 찾아볼 수 있도록 문제점을 개선하는 업무를 맡게 되었다.

개발의 방향을 잡기 위해 튜브는 우선 최근 화제가 되고 있는 "카카오 신입 개발자 공채" 관련 기사를 검색해보았다.

  • 카카오 첫 공채..'블라인드' 방식 채용
  • 카카오, 합병 후 첫 공채.. 블라인드 전형으로 개발자 채용
  • 카카오, 블라인드 전형으로 신입 개발자 공채
  • 카카오 공채, 신입 개발자 코딩 능력만 본다
  • 카카오, 신입 공채.. "코딩 실력만 본다"
  • 카카오 "코딩 능력만으로 2018 신입 개발자 뽑는다"

기사의 제목을 기준으로 "블라인드 전형"에 주목하는 기사와 "코딩 테스트"에 주목하는 기사로 나뉘는 걸 발견했다. 튜브는 이들을 각각 묶어서 보여주면 카카오 공채 관련 기사를 찾아보는 사용자에게 유용할 듯싶었다.

유사한 기사를 묶는 기준을 정하기 위해서 논문과 자료를 조사하던 튜브는 "자카드 유사도"라는 방법을 찾아냈다.

자카드 유사도는 집합 간의 유사도를 검사하는 여러 방법 중의 하나로 알려져 있다. 두 집합 A, B 사이의 자카드 유사도 J(A, B)는 두 집합의 교집합 크기를 두 집합의 합집합 크기로 나눈 값으로 정의된다.

예를 들어 집합 A = {1, 2, 3}, 집합 B = {2, 3, 4}라고 할 때, 교집합 A ∩ B = {2, 3}, 합집합 A ∪ B = {1, 2, 3, 4}이 되므로, 집합 A, B 사이의 자카드 유사도 J(A, B) = 2/4 = 0.5가 된다. 집합 A와 집합 B가 모두 공집합일 경우에는 나눗셈이 정의되지 않으니 따로 J(A, B)= 1로 정의한다.

자카드 유사도는 원소의 중복을 허용하는 다중집합에 대해서 확장할 수 있다. 다중집합 A는 원소 "1"을 3개 가지고 있고, 다중집합 B는 원소 "1"을 5개 가지고 있다고 하자. 이 다중집합의 교집합 A ∩ B는 원소 "1"을 min(3, 5)인 3개, 합집합 A ∪ B는 원소 "1"을 max(3, 5)인 5개 가지게 된다. 다중집합 A = {1, 1, 2, 2, 3}, 다중집합 B = {1, 2, 2, 4, 5}라고 하면, 교집합 A ∩ B = {1, 2, 2}, 합집합 A ∪ B = {1, 1, 2, 2, 3, 4, 5}가 되므로, 자카드 유사도 J(A, B) = 3/7, 약 0.42가 된다.

이를 이용하여 문자열 사이의 유사도를 계산하는데 이용할 수 있다. 문자열 "FRANCE"와 "FRENCH"가 주어졌을 때, 이를 두 글자씩 끊어서 다중집합을 만들 수 있다. 각각 {FR, RA, AN, NC, CE}, {FR, RE, EN, NC, CH}가 되며, 교집합은 {FR, NC}, 합집합은 {FR, RA, AN, NC, CE, RE, EN, CH}가 되므로, 두 문자열 사이의 자카드 유사도 J("FRANCE", "FRENCH") = 2/8 = 0.25가 된다.

제한사항

  • 입력으로는 str1str2의 두 문자열이 들어온다. 각 문자열의 길이는 2 이상, 1,000 이하이다.
  • 입력으로 들어온 문자열은 두 글자씩 끊어서 다중집합의 원소로 만든다. 이때 영문자로 된 글자 쌍만 유효하고, 기타 공백이나 숫자, 특수 문자가 들어있는 경우는 그 글자 쌍을 버린다. 예를 들어 "ab+"가 입력으로 들어오면, "ab"만 다중집합의 원소로 삼고, "b+"는 버린다.
  • 다중집합 원소 사이를 비교할 때, 대문자와 소문자의 차이는 무시한다. "AB"와 "Ab", "ab"는 같은 원소로 취급한다.

입출력

str1str2result
FRANCEfrench
handshakeshake hands
aa1+aa2AAAA12
E=M*C^2e=m*c^2

출력 형식

입력으로 들어온 두 문자열의 자카드 유사도를 출력한다. 유사도 값은 0에서 1 사이의 실수이므로, 이를 다루기 쉽도록 65536을 곱한 후에 소수점 아래를 버리고 정수부만 출력한다.

해결

문자열str1str2의 부분 문자열의 교집합 갯수 / 합집합 갯수 * 65536 의 정수값을 리턴하면 된다.

두 문자열을 toLowerCase() 로 소문자로 통일해준다.

getSubString()으로 길이가 2인 부분 문자열 ArrayList를 만든다.
부분 문자열에는 알파벳이 아닌 문자가 포함되지 않아야 하기에 그런 부분 문자열은 제외하여 추가한다.

부분 문자열 리스트인 str1Liststr2List 의 교집합 리스트와 합집합 리스트를 만들기위해 알파벳순으로 정렬해준다.

getIntersection() 은 교집합인 부분 문자열을 추출하기위한 메서드 이다.
정렬된 str1Liststr2List 을 동시에 탐색한다.
각각의 인덱스를 추적할수 있게 str1ListIdxstr2ListIdx를 따로 선언하고
두 리스트중 하나라도 종료될시 탐색은 끝난다.
각 인덱스가 가르키는 문자열인 str1str2를 비교하고 같으면 그 문자열을 추가후 두 인덱스를 증가하고
다르다면 두 문자열 str1str2중에서 사전순으로 더 앞쪽인 문자열의 인덱스를 증가시킨다.
이렇게 교집합 문자열을 모두 찾는다.

getUnion() 은 합집합 부분 문자열을 추출하는 메서드이다.
교집합과 같이 정렬된 str1Liststr2List 을 동시에 탐색하고 인덱스를 추적하는 str1ListIdxstr2ListIdx를 따로 선언한다.
합집합은 두 문자열이 같다면 그 문자열을 추가하고 두 인덱스를 증가,
다를 경우 사전상 앞쪽인 문자열을 추가하고 그 인덱스를 증가한다.

만약 반복이 끝난후 한쪽의 문자열이 전부 탐색되지 않았을 경우 체크한 인덱스부터 끝까지 부분 문자열 리스트를 탐색하여 합집합에 추가해 모든 합집합을 구한다.

코드

import java.util.ArrayList;
import java.util.Collections;

class Solution {
    public int solution(String str1, String str2) {
        int answer = 0;
        
        int NUMBER = 65536;
        String lowerStr1 = str1.toLowerCase();
        String lowerStr2 = str2.toLowerCase();

        ArrayList<String> str1List = new ArrayList<>();
        ArrayList<String> str2List = new ArrayList<>();

        getSubString(lowerStr1, str1List);
        getSubString(lowerStr2, str2List);

        Collections.sort(str1List);
        Collections.sort(str2List);

        ArrayList<String> intersectionList = new ArrayList<>();
        ArrayList<String> unionList = new ArrayList<>();

        getIntersection(intersectionList, str1List, str2List);
        getUnion(unionList, str1List, str2List);

        double J = ((double) intersectionList.size() / (double) unionList.size());
        if (unionList.size() == 0) J = 1.0D;
        answer = (int) (J * NUMBER);
        
        return answer;
    }
    
    private static void getIntersection(ArrayList<String> intersectionList, ArrayList<String> str1List, ArrayList<String> str2List) {
        int str1ListIdx = 0;
        int str2ListIdx = 0;

        for (; str1ListIdx < str1List.size() && str2ListIdx < str2List.size(); ) {
            String str1 = str1List.get(str1ListIdx);
            String str2 = str2List.get(str2ListIdx);

            if (str1List.get(str1ListIdx).equals(str2List.get(str2ListIdx))) {
                intersectionList.add(str1);
                str1ListIdx++;
                str2ListIdx++;
            } else {
                if (str1.compareTo(str2) < 0) str1ListIdx++;
                else str2ListIdx++;
            }
        }
    }

    private static void getUnion(ArrayList<String> unionList, ArrayList<String> str1List, ArrayList<String> str2List) {
        int str1ListIdx = 0;
        int str2ListIdx = 0;

        for (; str1ListIdx < str1List.size() && str2ListIdx < str2List.size(); ) {
            String str1 = str1List.get(str1ListIdx);
            String str2 = str2List.get(str2ListIdx);

            if (str1List.get(str1ListIdx).equals(str2List.get(str2ListIdx))) {
                unionList.add(str1);
                str1ListIdx++;
                str2ListIdx++;
            } else {
                if (str1.compareTo(str2) < 0) {
                    unionList.add(str1);
                    str1ListIdx++;
                } else {
                    unionList.add(str2);
                    str2ListIdx++;
                }
            }
        }

        for (; str1ListIdx < str1List.size(); str1ListIdx++) {
            String str1 = str1List.get(str1ListIdx);
            unionList.add(str1);
        }

        for (; str2ListIdx < str2List.size(); str2ListIdx++) {
            String str2 = str2List.get(str2ListIdx);
            unionList.add(str2);
        }

    }

    private static void getSubString(String lowerStr, ArrayList<String> strList) {
        for (int i = 0; i < lowerStr.length() - 1; i++) {
            String subStr = lowerStr.substring(i, i + 2);
            
            if ('a' <= subStr.charAt(0) && subStr.charAt(0) <= 'z' && 'a' <= subStr.charAt(1) && subStr.charAt(1) <= 'z') {
                strList.add(lowerStr.substring(i, i + 2));
            }
        }
    }
}
profile
저장소
post-custom-banner

0개의 댓글