[프로그래머스] Lv.0 - 문자열안에 문자열.py

문제

• 문자열 str1, str2가 매개변수로 주어집니다. str1 안에 str2가 있다면 1을 없다면 2를 return하도록 solution 함수를 완성해주세요.

제한사항

• 1 ≤ str1의 길이 ≤ 100
• 1 ≤ str2의 길이 ≤ 100
• 문자열은 알파벳 대문자, 소문자, 숫자로 구성되어 있습니다.

입출력 예시

• 입출력 예 1
  • "ab6CDE443fgh22iJKlmn1o" str1에 str2가 존재하므로 1을 return합니다.
• 입출력 예 2
  • "ppprrrogrammers" str1에 str2가 없으므로 2를 return합니다.
• 입출력 예 3
  • "AbcAbcA" str1에 str2가 없으므로 2를 return합니다.

코드

def solution(str1, str2):
    answer = 2
    if str2 in str1:
        answer = 1
    return answer

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