[프로그래머스] 문자열 내 p와 y의 개수 (Swift)

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

프로그래머스

 

 

풀이

1. for loop으로 s의 문자열을 하나하나씩 받아와 y,Y,s,S를 비교한다. 

2. y,Y와 같다면 ycount 를 1증가

3. p,P와 같다면 pcount 를 1증가

4.ycount 와 pcount가 같다면 true 반환 (0개일때에도 둘다 0이기에 true반환)

5.나머지는 false니까 return false

import Foundation

func solution(_ s:String) -> Bool
{
    var ycount = 0
    var pcount = 0
    for i in s{
        if "y" == i || "Y" == i{
            ycount += 1
        }
        if "p" == i || "P" == i{
            pcount += 1
        }
    }
    if ycount == pcount{
        return true
    }
    
    return false
}