[programmers] 문자열 내 p와 y의 개수

문제

접근

1. 문자열 내 p의 갯수와 y의 갯수 카운트
2. 같으면 true, 다르면 false 리턴

코드

func solution(_ s:String) -> Bool {
    var answer: Bool = false
    var pCount = 0
    var yCount = 0
    
    for character in s {
        if character == "p" || character == "P" {
            pCount += 1
        }else if character == "y" || character == "Y" {
            yCount += 1
        }
    }
    
    if pCount == yCount {
        answer = true
    }
    
    return answer
}

리팩토링

func solution(_ s:String) -> Bool {
    
    return s.lowercased().filter { $0 == "p" }.count == s.lowercased().filter { $0 == "y" }.count
}