숙제1 - 문자열 연습하기

문제

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

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

 

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

프로그래머스

 

내가 작성한 코드

function solution(str) {

    let text = str.toUpperCase();

    let countP = 0;

    let countY = 0;

    for (let i = 0; i < text.length; i++) {

        if (text[i] == 'P') countP++;

        if (text[i] == 'Y') countY++;

    }

    console.log(countP, countY);

    return countP == countY;

}

var str = 'pPoooyY';

solution(str);

 

정답

코딩테스트 문제를 풀 땐, 생각해 둔 머리 속에 있는 로직을 주석으로 STEP을 정해놓은 후에 만드는 걸 추천

1. 대문자로 통일
2. for문을 통해서 문자열의 요소 하나하나씩을 비교
   p, y와 비교: num을 늘리고 줄이면서 비교. num이 0이면 갯수가 같은거
3. 개수 체크(p와 y의 개수가 같은지를 체크)

function solution(s){

    var result = true;

    s = s.toUpperCase();

    var num = 0;

    for(var i = 0; i < s.length; i++){

      if(s[i] === 'P') num++; // p이면 갯수 더하기

      if(s[i] === 'Y') num--; // y이면 갯수 빼기

    }

    result = (num === 0); // p, y 갯수가 같으면 0 (p가 많으면 양수, 반대 음수)

 

    return result;

  }