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

문제 설명

---

문제 풀이

function solution(s){
    const arr = s.toLowerCase().split("")
    let answer
    let p = []
    let y = []
    for(let i = 0; i < arr.length; i++) {
        if(arr[i] === "p") {
            p.push("p")
        } else if (arr[i] === "y") {
            y.push("y")
        }
    }
    p.length === y.length ? answer = true : answer = false
    return answer
}

1. 문자열을 split으로 쪼갠 배열을 for루프를 돌림

 

2. i번째 배열요소가 p이면 p에 y면 y에 집어넣음

 

3. 길이비교 후 리턴

---

개선 사항

function solution(s){
   return s.toLowerCase().split("p").length === s.toLowerCase().split("y").length
}

1. split("p")를 이용하면 p에서 끊어짐으로써 배열의 길이를 추출해낼 수 있다.

 

2. 굳이 불리언값을 각각 배정하지않고 리턴해도 된다.