본문 바로가기
프로그래머스,백준/알고리즘

[프로그래머스] Level1 - 문자열 내 p와 y의 개수 (Python)

by z.1nee 2020. 10. 25.
SMALL

문제 설명

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

예를 들어 s가 pPoooyY면 true를 return하고 Pyy라면 false를 return합니다.

 

제한사항

  • 문자열 s의 길이 : 50 이하의 자연수
  • 문자열 s는 알파벳으로만 이루어져 있습니다.

입출력 예

 

s answer
pPoooyY true
Pyy false

 

입출력 예 설명

입출력 예 #1
'p'의 개수 2개, 'y'의 개수 2개로 같으므로 true를 return 합니다.

입출력 예 #2
'p'의 개수 1개, 'y'의 개수 2개로 다르므로 false를 return 합니다.


알고리즘

1. answer는 True로 초기화 

2. Collection 모듈의 Counter 클래스를 사용해 dictionary 형태로 알파벳 별 개수를 Count

* 이때 대소문자를 구별하지 않고 Count 해야하므로 s 리스트를 upper() 함수로 대문자로 변환해줌

3. 만약 'P'(키)의 값과 'Y'(키)의 값이 다르면 answer를 False로 변경

 

 

필요 함수 정리

대소문자 변환

  • upper()
    • 형식 : 리스트.upper()
    • 소문자를 대문자로 바꾸어 줌
    • 이미 대문자라면 원래 문자열을 반환
  • lower()
    • 형식 : 리스트.lower()
    • 대문자를 소문자로 바꾸어 줌
    • 이미 소문자라면 원래 문자열을 반환

Counter 클래스

  • Counter
    •  import  : from collections import Counter 
    • 형식 : Counter(리스트)
    • 집합에서 각 원소의 출현 횟수를 셈
    • Dictionary 형태로 반환

 

 

 

내 소스 코드

from collections import Counter

def solution(s):
    answer = True
    dics = Counter(s.upper())
    
    if dics['P'] != dics['Y'] : answer = False
        
    return answer
    

 

결과 

 

 

다른 사람 소스 코드

1. cout

def solution(s):
    return s.lower().count('p') == s.lower().count('y')

 

알고리즘은 동일하나 바로 문자열의 개수를 찾아주는 count 함수가 있었음

동일하면 True 동일하지 않으면 False 를 반환해줌

 

* count

형식 : 리스트.count('찾을 문자열')

괄호( ) 안에 찾고자 하는 값을 입력하면 함수를 사용한 변수 안에서 해당 값의 개수를 숫자로 반환

 

댓글