오답노트

프로그래머스 1단계 - 문자열 내 p와 y의 개수 본문

C,C++/코딩테스트

프로그래머스 1단계 - 문자열 내 p와 y의 개수

권멋져 2022. 5. 27. 15:18
https://programmers.co.kr/learn/courses/30/lessons/12916
 

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

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

programmers.co.kr

- 문제 파악

문자열 s가 주어지고 그 안에 p 의 개수와 y의 개수가 일치하면 true 그렇지 않으면 false를 출력하라

 

- 정답

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

bool solution(string s)
{
    bool answer = true;
    int pcnt = 0;
    int ycnt = 0;
    for(auto a : s)
    {
        if(a == 'p' || a== 'P')
        {
            pcnt++;
        }
        
        if(a == 'y' || a== 'Y')
        {
            ycnt++;
        }
    }
    
    if(pcnt != ycnt)
        answer = false;
    
    // [실행] 버튼을 누르면 출력 값을 볼 수 있습니다.
    //cout << "Hello Cpp" << endl;

    return answer;
}