[c++] 프로그래머스 - 문자열 내 p와 y의 개수( Level 1)

문제

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

 

코드

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

bool solution(string s)
{
    bool answer = true;

    int p_cnt=0,y_cnt=0;

    for(int i=0; i<s.length(); i++){
        if(s[i]=='p'||s[i]=='P') p_cnt++;
        else if(s[i]=='y'||s[i]=='Y') y_cnt++;
    }

    if(p_cnt==y_cnt) answer=true;
    else answer=false;

    return answer;
}