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

[문제]

코딩테스트 연습 - 문자열 내 p와 y의 개수 | 프로그래머스 스쿨 (programmers.co.kr)

프로그래머스

[문제 풀이]

p나 P 그리고 y나 Y를 더해봤을때 둘의 숫자가 같으면 true 아니면 false를 반환하자.

 

두개의 변수 pCount와 yCount를 만들어서 s를 돌면서 해당 알파벳들을 더해주고 마지막에 같은지 확인해주자.

[회고]

.

[코드]

import Foundation

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