[프로그래머스 Python] 코딩테스트 입문 Day 17 - 문자열, 수학, 조건문, 배열, 사칙연산

• 숫자 찾기
• n의 배수 고르기
• 자릿수 더하기
• OX퀴즈

---

1. 숫자 찾기

정수 num과 k가 매개변수로 주어질 때, num을 이루는 숫자 중에 k가 있으면 num의 그 숫자가 있는 자리 수를 return하고 없으면 -1을 return 하도록 solution 함수를 완성해보세요.

제한사항

• 0 < num < 1,000,000
• 0 ≤ k < 10
• num에 k가 여러 개 있으면 가장 처음 나타나는 자리를 return 합니다.

코드 작성

def solution(num, k):
    str_num = str(num)
    str_k = str(k)
    return -1 if str_k not in str_num else str_num.index(str_k) + 1

2. n의 배수 고르기

정수 n과 정수 배열 numlist가 매개변수로 주어질 때, numlist에서 n의 배수가 아닌 수들을 제거한 배열을 return하도록 solution 함수를 완성해주세요.

제한사항

• 1 ≤ n ≤ 10,000
• 1 ≤ numlist의 크기 ≤ 100
• 1 ≤ numlist의 원소 ≤ 100,000

코드 작성

def solution(n, numlist):
    answer = [num for num in numlist if num % n == 0]      
    return answer

3. 자릿수 더하기

정수 n이 매개변수로 주어질 때 n의 각 자리 숫자의 합을 return하도록 solution 함수를 완성해주세요

제한사항

• 0 ≤ n ≤ 1,000,000

코드 작성

def solution(n):
    nums = [int(x) for x in str(n)]
    return sum(nums)

4. OX퀴즈

덧셈, 뺄셈 수식들이 ‘X [연산자] Y = Z’ 형태로 들어있는 문자열 배열 quiz가 매개변수로 주어집니다. 수식이 옳다면 “O”를 틀리다면 “X”를 순서대로 담은 배열을 return하도록 solution 함수를 완성해주세요.

제한사항

• 연산 기호와 숫자 사이는 항상 하나의 공백이 존재합니다. 단 음수를 표시하는 마이너스 기호와 숫자 사이에는 공백이 존재하지 않습니다.
• 1 ≤ quiz의 길이 ≤ 10
• X, Y, Z는 각각 0부터 9까지 숫자로 이루어진 정수를 의미하며, 각 숫자의 맨 앞에 마이너스 기호가 하나 있을 수 있고 이는 음수를 의미합니다.
• X, Y, Z는 0을 제외하고는 0으로 시작하지 않습니다.
• -10,000 ≤ X, Y ≤ 10,000
• -20,000 ≤ Z ≤ 20,000
• [연산자]는 + 와 - 중 하나입니다.

코드 작성

def solution(quiz):
    answer = []
    s_quiz = [q.split(" ") for q in quiz]
    for l in s_quiz:
        print(l)
        if l[1] == '-':
            if int(l[0]) - int(l[2]) == int(l[-1]):
                answer.append("O")
            else:
                answer.append("X")
        else:
            if int(l[0]) + int(l[2]) == int(l[-1]):
                answer.append("O")
            else:
                answer.append("X")                
    return answer

👩🏻‍💻개인 공부 기록용 블로그입니다
오류나 틀린 부분이 있을 경우 댓글 혹은 메일로 따끔하게 지적해주시면 감사하겠습니다.