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

문제

• 어떤 문자열에 대해서 접미사는 특정 인덱스부터 시작하는 문자열을 의미합니다. 예를 들어, "banana"의 모든 접미사는 "banana", "anana", "nana", "ana", "na", "a"입니다.
  문자열 my_string과 is_suffix가 주어질 때, is_suffix가 my_string의 접미사라면 1을, 아니면 0을 return 하는 solution 함수를 작성해 주세요.

제한사항

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

입출력 예시

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

코드 1

def solution(my_string, is_suffix):
    answer = 0

    idx_word_list = []
    for idx in range(len(my_string)):
        idx_word = my_string[idx:]
        idx_word_list.append(idx_word)

    if is_suffix in idx_word_list:
        answer = 1
    return answer

코드 2

def solution(my_string, is_suffix):
    answer = 0
    if my_string.endswith(is_suffix):
        answer = 1
    return answer

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