[프로그래머스] 뉴스 클러스터링 Swift

프로그래머스 - 뉴스 클러스터링

나의 풀이

5번과 13번 통과 못하시는 분들은 A와 집합 B가 모두 공집합일 경우에는 나눗셈이 정의되지 않으니 따로 J(A, B) = 1 이 조건을 생각해주세요 !! A 합집합 B가 0일때는 65536을 return 해야 됩니다.

func solution(_ str1:String, _ str2:String) -> Int {
    let str1Arr = Array(str1)
    let str2Arr = Array(str2)
    
    var str1arr: [String] = []
    var str2arr: [String] = []
    // str1을 두 글자씩 합치기
    for i in 0..<str1Arr.count - 1 {
        if str1Arr[i].isLetter && str1Arr[i+1].isLetter{
            str1arr.append("\(str1Arr[i].uppercased())\(str1Arr[i+1].uppercased())")
        }
    }
    // str2를 두 글자씩 합치기
    for i in 0..<str2Arr.count - 1 {
        if str2Arr[i].isLetter && str2Arr[i+1].isLetter{
            str2arr.append("\(str2Arr[i].uppercased())\(str2Arr[i+1].uppercased())")
        }
    }
    
    var allCnt =  str1arr.count + str2arr.count
    var myCnt = 0
    // A 교집합 B 구하기
    for i in str1arr.indices{
        for j in str2arr.indices{
            if str1arr[i] == str2arr[j]{
                myCnt += 1
                str2arr.remove(at: j)
                break
            }
        }
    }
    // A 합집합 B = A + B - A 교집합 B
    allCnt -= myCnt
    
    if allCnt == 0{ // A와 집합 B가 모두 공집합일 경우에는 나눗셈이 정의되지 않으니 따로 J(A, B) = 1
        return 65536
    }
    else if myCnt == 0{
        return 0
    }
    else{
        return Int(Double(myCnt) / Double(allCnt) * 65536)
    }
}