[Python/코테공부] 코드 처리하기

[Python] 코드 처리하기

 

코드 처리하기 문제

 

내 풀이

사실 푸는데 애 좀 먹었다. 문제 이해를 잘못해서..

전체 문자열의 인덱스에 대해 홀짝을 계산했어야 하는데 split된 문자열에 대해 계산해서 오류가 나왔다.

그래서 고친 코드는 이거다.

def solution(code):  # code = "abc1abc1abc"
    answer = ''
    mode = 0
    idx = 0
    
    for c in code:
        if c== '1':
            mode = abs(mode-1)
            idx += 1
            continue
        elif (idx+mode)%2 == 0:
            answer += c
            idx +=1
        else:
            idx +=1
            
    return answer if len(answer) != 0 else "EMPTY"

 

테스트 결과

내 풀이 테스트 결과

 

풀리긴 했다만 시간이 좀 걸렸다.

 

다른 사람의 풀이

역시나 경이롭다.

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

 

잘라진 문자열을 다 이어 붙여서 인덱스를 2씩 증가시킨 것만 가져왔다.

인덱스 홀/짝 고려를 하지 않아도 된다니.. 애초에 이런 로직을 고려하고 만든 문제같다.

이런 생각은 전혀 하지 못했다..

더 열심히 공부해야겠다..!ㅠㅠ

 

테스트 결과

join을 썼을 때 결과

 

시간도 훠얼씬 빠르다.. 😥

 

*문제 출처 프로그래머스

https://school.programmers.co.kr/learn/courses/30/lessons/181932#

프로그래머스