프로그래머스 . 문자열 내 p와 y의 개수(Lv1)

 

 

 

- 파이썬

def solution(s):
    answer = True
    s = s.lower()
    sc = s.count('p')
    pc = s.count('y')
    
    if sc == pc:
        answer = True
    else:
        answer = False
    

    return answer

- C++

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

bool solution(string s)
{
    bool answer = true;
    int pc = 0;
    int yc = 0;
    
    for(int i=0;i < s.size();i++)
    {

        if(s[i] == 'p' || s[i] == 'P')
        {
            pc ++;
        }
        else if(s[i] == 'y' || s[i] == 'Y')
        {
            yc ++;
        }
    }
    
    if(pc != yc)
    {
        answer = false;
    }
    
    

    return answer;
}