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

문제

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

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

문제풀이

0부터 s의 크기만큼 반복하면서 p or P, y or Y를 찾으면 각각 변수에 +1을 해준다.

p와 y가 다르면 answer = false이고, 같으면 answer = true가 된다.

 

코드

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

bool solution(string s)
{
    bool answer = true;
    int idx=0;
    int p=0, y=0;
    for(idx=0;idx<s.size();++idx){
        if(s[idx]=='p'||s[idx]=='P')
           p++;
        else if(s[idx]=='y'||s[idx]=='Y')
            y++;
    }    
    if(p!=y) answer=false;

    return answer;
}