[Swift 알고리즘] 체육복

장주명·2021년 5월 11일

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

import Foundation

func solution(_ n:Int, _ lost:[Int], _ reserve:[Int]) -> Int {
    
    let lostUnifrom = lost.filter {!reserve.contains($0)}
    var reserveUnifrom = reserve.filter {!lost.contains($0)}
    var answer = n - lostUnifrom.count
    
    for i in 0..<lostUnifrom.count {
        for j in 0..<reserveUnifrom.count {
            if abs(lostUnifrom[i] - reserveUnifrom[j]) == 1 {
                answer += 1
                reserveUnifrom.remove(at: j)
                break
            }
        }
        
    }
    
    return answer
}
profile
flutter & ios 개발자

0개의 댓글