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

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

코딩테스트 연습 - 문자열 내 p와 y의 개수

문자열 안의 p와 y의 개수를 카운트해서 같다면 true, 다르다면 false 를 리턴합니다.

 

개수를 카운트해서 더해준 후, 마지막에 비교해서 다르면 false 같으면 true를 리턴합니다.

#include <string>
#include <iostream>
using namespace std;

bool solution(string s)
{
    bool answer = true;
    int pCount = 0;     // p 개수 카운트
    int yCount = 0;     // y 개수 카운트
    
    for(int i=0; i<s.length(); i++)
    {
        if(s[i] == 'p' || s[i] == 'P')  // p개수 카운트
        {
            pCount++;
        }
        if(s[i] == 'y' || s[i] == 'Y')  // y개수 카운트
        {
            yCount++;
        }
    }
    
    if(yCount != pCount)    // 다르다면 false
        answer = false;
    
    return answer;
}