[프로그래머스] Lv.0 - 0 떼기.py

문제

• 정수로 이루어진 문자열 n_str이 주어질 때, n_str의 가장 왼쪽에 처음으로 등장하는 0들을 뗀 문자열을 return하도록 solution 함수를 완성해주세요.

제한사항

• 2 ≤ n_str ≤ 10
• n_str이 "0"으로만 이루어진 경우는 없습니다.

입출력 예시

• 입출력 예 1
  • "0010"의 가장 왼쪽에 연속으로 등장하는 "0"을 모두 제거하면 "10"이 됩니다.
• 입출력 예 2
  • "854020"는 가장 왼쪽에 0이 없으므로 "854020"을 return합니다.

코드 1

def solution(n_str):
    answer = n_str
    count = 0
    for val in n_str:
        if val == "0":
            count += 1
        else:
            answer = n_str[count:]
            break
    return answer

코드 2

def solution(n_str):
    answer = n_str.lstrip("0")
    return answer

출처: 프로그래머스 코딩 테스트 연습, https://school.programmers.co.kr/learn/challenges