[2018 카카오 블라인드 채용] 뉴스 클러스터링

문제

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

프로그래머스

 

소스코드

import java.util.PriorityQueue;
import java.util.stream.Collectors;
import java.util.stream.IntStream;

class Solution {
    public int solution(String str1, String str2) {
        PriorityQueue<String> q1 = setQueue(str1);
        PriorityQueue<String> q2 = setQueue(str2);

        if(q1.isEmpty() && q2.isEmpty()) return 65536;

        double inter = 0, union = 0;
        while(!q1.isEmpty() && !q2.isEmpty()) {
            String s1 = q1.poll();
            String s2 = q2.poll();

            int cmp = s1.compareTo(s2);
            if(cmp == 0) inter++;
            else if(cmp < 0) q2.offer(s2);
            else q1.offer(s1);
            union++;
        }

        union += (q1.size() + q2.size());
        return (int)((inter/union)*65536);
    }
    
    public PriorityQueue<String> setQueue(String str) {
        return IntStream.rangeClosed(0, str.length()-2)
                .mapToObj((n) -> str.substring(n, n+2).toUpperCase())
                .filter(s -> s.matches("[a-zA-Z]{2}"))
                .collect(Collectors.toCollection(PriorityQueue::new));
	}
}

 

회고

- 빠른 비교를 위해 문자열 정렬이 필요했음.