프로그래머스-코딩테스트/Lv.1
이상한 문자 만들기
stem_sw
2024. 9. 6. 16:00
프로그래머스
코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.
programmers.co.kr
문제
문자열 s는 한 개 이상의 단어로 구성되어 있습니다. 각 단어는 하나 이상의 공백문자로 구분되어 있습니다. 각 단어의 짝수번째 알파벳은 대문자로, 홀수번째 알파벳은 소문자로 바꾼 문자열을 리턴하는 함수, solution을 완성하세요.
제한 사항
- 문자열 전체의 짝/홀수 인덱스가 아니라, 단어(공백을 기준)별로 짝/홀수 인덱스를 판단해야합니다.
- 첫 번째 글자는 0번째 인덱스로 보아 짝수번째 알파벳으로 처리해야 합니다.
코드
def solution(s):
cnt = 1
answer = ''
for i in s:
if i == ' ':
answer += i
cnt=1
elif cnt == 1:
answer += i.upper()
cnt -= 1
elif cnt == 0:
answer += i.lower()
cnt += 1
return answer
- 공백인지 확인 + 홀수번째, 짝수번째 구분후 대소문자로 변경
def solution(s):
answer = []
for i in s.split(' '):
answer.append([m.lower() if n%2 else m.upper() for n,m in enumerate(i)])
return ' '.join([''.join(j) for j in answer])
- 중첩리스트를 문자열로 만들어줌. list comprehension으로 join여러번
- enumerate()로 요소별 인덱스를 부여해 홀짝 구분
노트
공백문자에 upper()
print('a'+' '.upper()+'a')
# a a
- upper()와 lower()모두 잘 적용됨
공백문자열 추가 append VS extend
s = "try hello world"
answer = []
for i in s.split(' '):
answer.append([m.upper() if n%2 else m.lower() for n,m in enumerate(i)])
print(answer)
answer = []
for i in s.split(' '):
answer.extend([m.upper() if n%2 else m.lower() for n,m in enumerate(i)])
print(answer)
# output
[['t', 'R', 'y'], ['h', 'E', 'l', 'L', 'o'], [], [], ['w', 'O', 'r', 'L', 'd']]
['t', 'R', 'y', 'h', 'E', 'l', 'L', 'o', 'w', 'O', 'r', 'L', 'd']
- extend는 공백문자열을 요소로 추가하지 못했다.
- split()을 쓰면 여러개의 공백문자열을 하나로 취급해 분리하지만, split(' ')로 설정해두면 띄어쓰기 한칸 만을 구분자로 구분