프로그래밍응용/오답노트
프로그래머스, 문자열 내 p와 y의 개수(C++, Python)
photoner
2021. 2. 28. 21:30
728x90
반응형
[ C++ ]
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
|
#include <string>
#include <iostream>
using namespace std;
bool solution(string s)
{
int pcnt=0, ycnt=0;
for(auto ch : s) // 모든 s의 문자들을 순회하면서
if(toupper(ch) == 'P') // 대문자화 한 문자가 P라면
pcnt++; // p 카운팅
else if(toupper(ch) == 'Y') // 대문자화 한 문자가 Y라면
ycnt++; // y 카운팅
return (pcnt == ycnt)? true : false;
}
|
cs |
[ Python ]
1
2
3
|
def solution(s):
s = s.upper() # 모두 대문자로 변경하고 '갱신'
return True if s.count('P') == s.count('Y') else False # Count를 통해 수를 세고, 삼항 연산을 통해 각 수를 비교
|
cs |
string 에서 count 함수는 해당 문자가 string 에 몇 개 들어있는가를 반환!
728x90
반응형