프로그래머스 코딩테스트 연습 [코드 처리하기]

문제 설명

문자열 code가 주어집니다.

code를 앞에서부터 읽으면서 만약 문자가 "1"이면 mode를 바꿉니다.

mode에 따라 code를 읽어가면서 문자열 ret을 만들어냅니다.

 

mode는 0과 1이 있으며, Idx를 0부터 code의 길이 - 1까지 1씩 키워나가면서 code[idx]의 값에 따라 다음과 같이 행동합니다.

 

• mode가 0일 때
  • code[idx]가 "1"이 아니면 idx가 짝수일 때만 ret의 맨 뒤에 code[idx]를 추가합니다.
  • code[idx]가 "1"이면 mode를 0에서 1로 바꿉니다.
• mode가 1일 때
  • code[idx]가 "1"이 아니면 idx가 홀수일 때만 ret의 맨 뒤에 code[idx]를 추가합니다.
  • code[idx]가 "1"이면 mode를 1에서 0으로 바꿉니다.

 

문자열 code를 통해 만들어진 문자열 ret를 return 하는 solution 함수를 완성해 주세요.

단, 시작할 때 mode는 0이며, return 하려는 ret가 만약 빈 문자열이라면 대신 "EMPTY"를 return 합니다.

 

제한사항

• 1 <= code의 길이 <= 100,000
  • code는 알파벳 소문자 또는 "1"로 이루어진 문자열입니다.

입출력 예

code	result
"abc1abc1abc"	"acbac"

 

입출력 예 설명

입출력 예 #1

• code의 각 인덱스 i에 따라 다음과 같이 mode와 ret가 변합니다.

i	code[i]	mode	ret
0	"a"	0	"a"
1	"b"	0	"a"
2	"c"	0	"ac"
3	"1"	1	"ac"
4	"a"	1	"ac"
5	"b"	1	"acb"
6	"c"	1	"acb"
7	"1"	0	"acb"
8	"a"	0	"acba"
9	"b"	0	"acba"
10	"c"	0	"acbac"

따라서 "acbac"를 return 합니다.

 

해결

내 풀이

def solution(code):
    answer = ''
    mode = 0
    for i in range(len(code)):
        if mode == 0:
            if code[i] != '1':
                if i % 2 == 0:
                    answer = answer + code[i]
            elif code[i] == '1':
                mode = 1
        elif mode == 1:
            if code[i] != '1':
                if i % 2 == 1:
                    answer = answer + code[i]
            elif code[i] == '1':
                mode = 0
    if answer == "":
        answer = "EMPTY"
    return answer

 

다른 풀이

다른 풀이 #1

def solution(code):
    return "".join(code.split("1"))[::2] or "EMPTY"

 

* "".join(code.split("1"))[::2] or "EMPTY"

 

1을 기준으로 자르고

합쳐서 (join)

인덱스가 짝수인 값(간격이 2)만 고르기

값이 없다면 "EMPTY"

 

다른 풀이 #2

def solution(code):
    answer = ''

    mode = 0
    for i in range(len(code)):
        if code[i] == '1':
            mode ^= 1
        else:
            if i % 2 == mode:
                answer += code[i]

    return answer if answer else 'EMPTY'

 

* mode ^= 1

* if i % 2 == mode:

 

새로 알게 된 점

^= : 기존값과 비교해서 바꾸는 연산자 (스위칭)

str.split("a") : str 을 "a"를 기준으로 나눔