[프로그래머스 스위프트] 문자열 내 p와 y의 개수

문제 요약

1. p, y, P, Y이 s에 포함되어 있지 않으면 true를 리턴

2. 다음으로 p와 y의 갯수를 센다. 서로 동일하면 true와 아니면 false를 리턴한다.

 

해답

import Foundation

func solution(_ s:String) -> Bool {
    var pCnt = 0
    var yCnt = 0
    if !s.contains("p") && !s.contains("y") && !s.contains("P") && !s.contains("Y") {
        return true
    }
    for i in s {
        if i == "p" || i == "P" {
            pCnt += 1
        } else if i == "Y" || i == "y" {
            yCnt += 1
        }
    }
    if pCnt == yCnt {
        return true
    } else {
        return false
    }
}