본문 바로가기
코딩테스트

프로그래머스 기초 Day 23 파이썬

by 피갓자 2025. 4. 24.

프로그래머스
프로그래머스

부분 문자열

문제 설명

어떤 문자열 A가 다른 문자열 B안에 속하면 A를 B의 부분 문자열이라고 합니다.

예를 들어 문자열 "abc"는 문자열 "aabcc"의 부분 문자열입니다.

문자열 str1과 str2가 주어질 때, str1이 str2의 부분 문자열이라면 1을 부분 문자열이 아니라면 0을 return하도록 solution 함수를 완성해주세요.

제한사항

  • 1 ≤ str1 ≤ str2 ≤ 20
  • str1과 str2는 영어 소문자로만 이루어져 있습니다.

입출력 예

str1 str2 result
"abc" "aabcc" 1
"tbt" "tbbttb" 0

입출력 예 설명

입출력 예 #1

  • 본문과 동일합니다.

입출력 예 #2

  • "tbbttb"에는 "tbt"가 없으므로 0을 return합니다.

solution.py

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

꼬리 문자열

문제 설명

문자열들이 담긴 리스트가 주어졌을 때, 모든 문자열들을 순서대로 합친 문자열을 꼬리 문자열이라고 합니다.

꼬리 문자열을 만들 때 특정 문자열을 포함한 문자열은 제외시키려고 합니다.

예를 들어 문자열 리스트 ["abc", "def", "ghi"]가 있고 문자열 "ef"를 포함한 문자열은 제외하고 꼬리 문자열을 만들면 "abcghi"가 됩니다.

문자열 리스트 str_list와 제외하려는 문자열 ex가 주어질 때, str_list에서 ex를 포함한 문자열을 제외하고 만든 꼬리 문자열을 return하도록 solution 함수를 완성해주세요.

제한사항

  • 2 ≤ str_list의 길이 ≤ 10
  • 1 ≤ str_list의 원소의 길이 ≤ 10
  • 1 ≤ ex의 길이 ≤ 5

입출력 예

str_list ex result
["abc", "def", "ghi"] "ef" "abcghi"
["abc", "bbc", "cbc"] "c" ""

입출력 예 설명

입출력 예 #1

  • 본문과 동일합니다.

입출력 예 #2

  • 리스트 안의 모든 문자열이 "c"를 포함하므로 빈 문자열을 return합니다.

solution.py

def solution(str_list, ex):
    answer = ''
    for i in str_list:
        if ex not in i:
            answer += i
    return answer

정수 찾기

문제 설명

정수 리스트 num_list와 찾으려는 정수 n이 주어질 때, num_list안에 n이 있으면 1을 없으면 0을 return하도록 solution 함수를 완성해주세요.

제한사항

  • 3 ≤ num_list의 길이 ≤ 100
  • 1 ≤ num_list의 원소 ≤ 100
  • 1 ≤ n ≤ 100

입출력 예

num_list n result
[1, 2, 3, 4, 5] 3 1
[15, 98, 23, 2, 15] 20 0

입출력 예 설명

입출력 예 #1

  • [1, 2, 3, 4, 5] 안에 3이 있으므로 1을 return합니다.

입출력 예 #2

  • [15, 98, 23, 2, 15] 안에 20이 없으므로 0을 return합니다.

solution.py

def solution(num_list, n):
    answer = 0
    if n in num_list:
        answer = 1
    else:
        answer = 0
    return answer

주사위 게임 1

문제 설명

1부터 6까지 숫자가 적힌 주사위가 두 개 있습니다.

두 주사위를 굴렸을 때 나온 숫자를 각각 ab라고 했을 때 얻는 점수는 다음과 같습니다.

  • a와 b가 모두 홀수라면 a^2 + b^2 점을 얻습니다.
  • a와 b 중 하나만 홀수라면 2 × (a + b) 점을 얻습니다.
  • a와 b 모두 홀수가 아니라면 |a - b| 점을 얻습니다.

두 정수 a와 b가 매개변수로 주어질 때, 얻는 점수를 return 하는 solution 함수를 작성해 주세요.

제한사항

  • a와 b는 1 이상 6 이하의 정수입니다.

입출력 예

a b result
3 5 34
6 1 14
2 4 2

입출력 예 설명

입출력 예 #1

  • 예제 1번에서 두 주사위 숫자가 모두 홀수이므로 32 + 52 = 9 + 25 = 34점을 얻습니다. 따라서 34를 return 합니다.

입출력 예 #2

  • 예제 2번에서 두 주사위 숫자 중 하나만 홀수이므로 2 × (6 + 1) = 2 × 7 = 14점을 얻습니다. 따라서 14를 return 합니다.

입출력 예 #3

  • 예제 3번에서 두 주사위 숫자가 모두 홀수가 아니므로 |2 - 4| = |-2| = 2점을 얻습니다. 따라서 2를 return 합니다.

solution.py

def solution(a, b):
    if a % 2 == 1 and b % 2 == 1:
        return a ** 2 + b ** 2
    elif (a % 2 == 1 and b % 2 == 0) or (a % 2 == 0 and b % 2 == 1):
        return 2 * (a + b)
    else:
        return abs(a - b)

날짜 비교하기

문제 설명

정수 배열 date1과 date2가 주어집니다.

두 배열은 각각 날짜를 나타내며 [year, month, day] 꼴로 주어집니다.

각 배열에서 year는 연도를, month는 월을, day는 날짜를 나타냅니다.

만약 date1이 date2보다 앞서는 날짜라면 1을, 아니면 0을 return 하는 solution 함수를 완성해 주세요.

제한사항

  • date1의 길이 = date2의 길이 = 3
    • 0 ≤ year ≤ 10,000
    • 1 ≤ month ≤ 12
    • day는 month에 따라 가능한 날짜로 주어집니다.

입출력 예

date1 date2 result
[2021, 12, 28] [2021, 12, 29] 1
[1024, 10, 24] [1024, 10, 24] 0

입출력 예 설명

입출력 예 #1

  • date1이 date2보다 하루 앞서기 때문에 1을 return 합니다.

입출력 예 #2

  • date1과 date2는 날짜가 서로 같으므로 date1이 더 앞서는 날짜가 아닙니다. 따라서 0을 return 합니다.

solution.py

def solution(date1, date2):
    if date1[0] < date2[0]:
        return 1
    elif date1[0] == date2[0]:
        if date1[1] < date2[1]:
            return 1
        elif date1[1] == date2[1]:
            if date1[2] < date2[2]:
                return 1
    return 0