[프로그래머스] Lv.0 - 접두사인지 확인하기.py

문제

• 어떤 문자열에 대해서 접두사는 특정 인덱스까지의 문자열을 의미합니다. 예를 들어, "banana"의 모든 접두사는 "b", "ba", "ban", "bana", "banan", "banana"입니다.
  문자열 my_string과 is_prefix가 주어질 때, is_prefix가 my_string의 접두사라면 1을, 아니면 0을 return 하는 solution 함수를 작성해 주세요.

제한사항

• 1 ≤ my_string의 길이 ≤ 100
• 1 ≤ is_prefix의 길이 ≤ 100
• my_string과 is_prefix는 영소문자로만 이루어져 있습니다.

입출력 예시

• 입출력 예 1
  • 예제 1번에서 is_prefix가 my_string의 접두사이기 때문에 1을 return 합니다.
• 입출력 예 2
  • 예제 2번에서 is_prefix가 my_string의 접두사가 아니기 때문에 0을 return 합니다.
• 입출력 예 3
  • 예제 3번에서 is_prefix가 my_string의 접두사가 아니기 때문에 0을 return 합니다.
• 입출력 예 4
  • 예제 4번에서 is_prefix가 my_string의 접두사가 아니기 때문에 0을 return 합니다.

코드 1

def solution(my_string, is_prefix):
    answer = 0
    prefix_list = []
    for idx in range(len(my_string)):
        prefix = my_string[:-idx]
        prefix_list.append(prefix)

    if is_prefix in prefix_list:
        answer = 1
    return answer

코드 2

def solution(my_string, is_prefix):
    answer = 0
    if my_string.startswith(is_prefix):
        answer = 1
    return answer

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