[Swift Algorithm] 문자열 내 p와 y의 개수 Programmers(Lv.1)

문제 설명

대문자와 소문자가 섞여있는 문자열 s가 주어집니다. s에 'p'의 개수와 'y'의 개수를 비교해 같으면 True, 다르면 False를 return 하는 solution를 완성하세요. 'p', 'y' 모두 하나도 없는 경우는 항상 True를 리턴합니다. 단, 개수를 비교할 때 대문자와 소문자는 구별하지 않습니다.

예를 들어 s가 "pPoooyY"면 true를 return하고 "Pyy"라면 false를 return합니다.

 

제한 조건

• 문자열 s의 길이 : 50 이하의 자연수
• 문자열 s는 알파벳으로만 이루어져 있습니다.

 

입출력 예

s	answer
"pPoooyY"	true
"Pyy"	false

 

내 문제 풀이

func solution(_ s:String) -> Bool
{
    var countP = 0
    var countY = 0
    
    for i in s.lowercased() {
        if i == "p" {
            countP += 1
        }
        else if i == "y" {
            countY += 1
        }
    }

    return countP == countY
}

• 먼저 주어진 문자열 s를 모두 소문자로 변경해주고, p나 y가 존재하는지 반복문으로 확인했다.
• return 문에서는 ==를 사용해 바로 Bool 값으로 리턴되도록 해주었다.

 

인상적인 풀이

import Foundation

func solution(_ s:String) -> Bool {
    let string = s.lowercased()
    return string.components(separatedBy: "p").count == string.components(separatedBy: "y").count
}

• components를 사용하여 p를 기준으로 나눈 배열의 요소 개수, y를 기준으로 나눈 배열의 요소 개수를 비교했다.

 

인상적인 풀이 하나 더,!

import Foundation

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

• 모두 소문자로 변경 후, filter 사용하여 p와 y의 개수 확인하고 비교해주었다.

 

💡 피드백

• 나도 나름 잘 풀었다고 생각했는데, 같은 로직인데도 filter를 사용하니 훨씬 간결해졌다,,!
• components를 이런 문제에도 사용할 수 있다는 것이 신기하다. 함수 사용에 정답은 없다,, 다양하게 함수를 사용할 수 있게 연습 많이 해야겠다.

 

---

 

🖍 [ 추가 ] 1주일 후 다시 풀어보기

func solution(_ s:String) -> Bool
{
    var pCount = 0
    var yCount = 0
    
    for i in s {
        if Array("pP").contains(i) {
            pCount += 1
        }
        else if Array("yY").contains(i) {
            yCount += 1
        }
    }
    
    return pCount == yCount
}

• if 문에서 바로 배열 pP, yY를 만들고, contains를 사용해 문자열을 검사했다.
• 효율적인 방법은 아닌듯 하다.ㅜ 위의 코드들 보면서 복습 많이 해야겠다.

 

---

 

문제

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