[Programmers] [1차] 뉴스 클러스터링 (JAVA)

프로그래머스

 

풀이

 

SSEULGRAMERS의 경우, [SS, SE, EU UL, LG, GR, RA, AM, ME, ER, RS]의 집합을 가지고

PROGRAMMERS의 경우 [PR, RO, OG, GR, RA, AM, MM, ME, ER, RS]의 집합을 가진다.

 

둘의 교집합은 [GR, RA, AM, ME, ER, RS]이고

둘의 합집합은 [SS, SE, EU, UL, LG, GR, PR, RO, OG, GR, RA, AM, MM, ME, ER, RS] 이다.

 

무조건 2음절로 나누어 교집합 합집합을 확인하기 때문에 Char를 활용하면 될 것 같다 생각했다.

AA 부터 ZZ까지 최대 26*26개의 경우의 수만 나오기 때문에

카운팅 배열을 활용해서

 

arr1과 arr2를 비교했을 때, 카운팅이 더 큰쪽이 합집합 카운팅이 더 작은 쪽이 교집합 요소 개수.

 

반례가 있는데, 공집합일 때는 0을 출력해 줘야한다.

반례

공집합일 때
answer = 0;

 

전체코드

import java.io.*;
import java.util.*;

class Solution {
    static int arr1[] = new int[706];
    static int arr2[] = new int[706];
    static int sum = 0;
    static int both = 0;
    public int solution(String str1, String str2) {
        
        str1 = str1.toUpperCase();
        str2 = str2.toUpperCase();
        
        Split(str1, arr1);
        Split(str2, arr2);
        
        Check();
        
        int answer = 0;
        
        if(sum == 0){
            answer = 1*65536;
        }else if(both == 0){
            answer = 0;
        }else{
            answer = 65536*both/sum;
        }
        return answer;
    }
    static void Check(){
        for(int i=0; i<706; i++){
            if(arr1[i]!=0 || arr2[i]!=0){
                sum += Math.max(arr1[i], arr2[i]);
                both += Math.min(arr1[i], arr2[i]);
            }
        }
    }
    static void Split(String str, int arr[]){
        for(int i=0; i<str.length()-1; i++){
            int tmp1 = (int)str.charAt(i) - 65;
            int tmp2 = (int)str.charAt(i+1) - 65;
            if(tmp1 >= 0 && tmp1 <= 26 && tmp2 >= 0 && tmp2 <= 26){
                int num = tmp1*26 + tmp2;
                arr[num]++;
            }else{
                continue;
            }
        }
    }
}