[프로그래머스 Python] 코딩 기초 트레이닝 Day 4 - 연산, 조건문

---

1. n의 배수

정수 num과 n이 매개 변수로 주어질 때, num이 n의 배수이면 1을 return n의 배수가 아니라면 0을 return하도록 solution 함수를 완성해주세요.

제한사항

• 2 ≤ num ≤ 100
• 2 ≤ n ≤ 9

코드 작성

def solution(num, n):
    return 1 if num % n == 0 else 0

2. 공배수

정수 number와 n, m이 주어집니다. number가 n의 배수이면서 m의 배수이면 1을 아니라면 0을 return하도록 solution 함수를 완성해주세요.

제한사항

• 10 ≤ number ≤ 100
• 2 ≤ n, m < 10

코드 작성

def solution(number, n, m):
    return 1 if number % n == 0 and number % m == 0 else 0

3. 홀짝에 따라 다른 값 반환하기

양의 정수 n이 매개변수로 주어질 때, n이 홀수라면 n 이하의 홀수인 모든 양의 정수의 합을 return 하고 n이 짝수라면 n 이하의 짝수인 모든 양의 정수의 제곱의 합을 return 하는 solution 함수를 작성해 주세요.

제한사항

• 1 ≤ n ≤ 100

코드 작성 (1)

def solution(n):
    return sum([n**2 for n in range(2, n+1, 2)]) if n % 2 == 0 else sum([n for n in range(1, n+1, 2)])

코드 작성 (2)

def solution(n):
    if n % 2 == 0:
        return sum([n**2 for n in range(2, n+1, 2)])
    return sum([n for n in range(1, n+1, 2)])

4. 조건 문자열

문자열에 따라 다음과 같이 두 수의 크기를 비교하려고 합니다.

• 두 수가 n과 m이라면
  • ”>”, “=” : n >= m
  • ”<”, “=” : n <= m
  • ”>”, “!” : n > m
  • ”<”, “!” : n < m

두 문자열 ineq와 eq가 주어집니다. ineq는 “<”와 “>”중 하나고, eq는 “=”와 “!”중 하나입니다. 그리고 두 정수 n과 m이 주어질 때, n과 m이 ineq와 eq의 조건에 맞으면 1을 아니면 0을 return하도록 solution 함수를 완성해주세요.

제한사항

• 1 ≤ n, m ≤ 100

코드 작성

def solution(ineq, eq, n, m):
    if ineq == '>':
        if eq == '=':
            return 1 if n >= m else 0
        else:
            return 1 if n > m else 0
    
    if eq == '=':
        return 1 if n <= m else 0
    else:
        return 1 if n < m else 0

5. flag에 따라 다른 값 반환하기

두 정수 a, b와 boolean 변수 flag가 매개변수로 주어질 때, flag가 true면 a + b를 false면 a - b를 return 하는 solution 함수를 작성해 주세요.

제한사항

• -1000 ≤ a, b ≤ 1,000

코드 작성

def solution(a, b, flag):
    return a + b if flag else a - b

6. 문제 출처

• n의 배수
• 공배수
• 홀짝에 따라 다른 값 반환하기
• 조건 문자열
• flag에 따라 다른 값 반환하기

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