2018 카카오 신입 뉴스 클러스터링 level 2 - java

 

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

두 글자씩 끊어서 다중집합을 만드는 stringSlice 메서드를 만들어서 Map을 리턴해주었다.

위처럼 다중집합을 만들어준 뒤, 문제에서 제시된대로 자카드 유사도를 구해주면 마무리된다.

import java.util.HashMap;
import java.util.Map;

class Solution {
    public int solution(String str1, String str2) {
        final int VALUE = 65536;
        str1 = str1.toLowerCase().replaceAll("[^a-z]"," ");
        str2 = str2.toLowerCase().replaceAll("[^a-z]"," ");
        Map<String,Integer> A = stringSlice(str1);
        Map<String,Integer> B = stringSlice(str2);

        int intersectionCount = 0; // 교집합의 개수
        int unionCount = 0; // 합집합의 개수
        for(Map.Entry<String, Integer> entry : B.entrySet()) {
            String str = entry.getKey();
            int count = entry.getValue();

            // 없다면
            if(A.get(str) == null) {
                unionCount += count;
            } else if(A.get(str) >= count) {
                intersectionCount += count;
                unionCount += A.get(str);
            } else if(A.get(str) < count){
                intersectionCount += A.get(str);
                unionCount += count;
            }
        }
        for(Map.Entry<String, Integer> entry : A.entrySet()) {
            String str = entry.getKey();
            int count = entry.getValue();
            System.out.println("key : " + str);
            System.out.println(B.get(str));
            // 없다면
            if(B.get(str) == null) {
                unionCount += count;
            }
        }
        // System.out.println("intersection : " + intersectionCount);
        // System.out.println("union : " + unionCount);
        if(intersectionCount == 0 && unionCount == 0){
            return VALUE;
        }
        double jacard = (double) intersectionCount / (double) unionCount;
        return (int) (VALUE * jacard);
    }

    private Map<String,Integer> stringSlice(String str) {
        Map<String,Integer> result = new HashMap<>();

        for(int i = 0 ;i<str.length()-1;i++) {
            String string = str.substring(i,i+2);
            if(string.contains(" ")) {
                continue;
            }
            result.put(string, result.getOrDefault(string,0) + 1);
        }

        return result;
    }
}