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

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

 

 

 문제 설명

 

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

프로그래머스

 

 

 

 나의 풀이

 

처음에는 Set 자료구조를 이용해서 풀어보려고 했는데 중복도 허용하기 때문에 배열을 이용했다.

먼저 allSatisfy메서드를 이용해서 문자열이 전부 문자인 경우에만 담아준 뒤에 첫번째 배열을 탐색하면서 두번째 배열에 값이 담겨있다면 제거, 아니면 넘어가는 형식으로 만들었다.

 

그리고 이 문제의 낚시포인트(?)는 집합 A와 집합 B가 공집합일 때는 1을 리턴하는 것인데 이는 교집합도 0이고 합집합도 0인 경우를 말한다. 그래서 이 부분을 분리해서 풀어줘야한다.

 

import Foundation
func solution(_ str1:String, _ str2:String) -> Int {
    var resultArr1: [String] = []
    var resultArr2: [String] = []
    
    var filteredArr1 = Array(str1.uppercased()).map{ String($0)}
    var filteredArr2 = Array(str2.uppercased()).map{ String($0)}
    var interCount = 0
    
    for (i, val) in filteredArr1.enumerated() {
        if i == filteredArr1.count-1 {
            break
        } 
        let compositionStr = val + filteredArr1[i+1] 
        if compositionStr.allSatisfy({ $0.isLetter }){
            resultArr1.append(compositionStr)
        }
    }
    
    for (i, val) in filteredArr2.enumerated() {
        if i == filteredArr2.count-1 {
            break
        } 
        let compositionStr = val + filteredArr2[i+1] 
        
        if compositionStr.allSatisfy({ $0.isLetter }){
            resultArr2.append(compositionStr)
        }
    }
    
    for x in resultArr1 {
        if resultArr2.contains(x) {
            resultArr1.remove(at: resultArr1.firstIndex(of: x)!)
            resultArr2.remove(at: resultArr2.firstIndex(of: x)!)
            interCount += 1
        } 
    }
    let sum = interCount + resultArr1.count + resultArr2.count
    
    return sum == 0 ? 65536 : Int((Double(interCount)  / Double(sum)) * 65536)
}