[Python] 프로그래머스 코딩테스트 입문 Lv.0 문제 모음 (1)

SQL 배우다보니 파이썬을 어느새 다 잊어가고 있기 때문에 ..

처음으로 돌아가지 않기 위한 최소한의 노력 모음 ❕

 

문제 1~5. 두 수의 합 / 두 수의 곱 / 두 수의 차 / 몫 구하기 / 나머지 구하기

1. 정수 num1과 num2가 주어질 때, num1과 num2의 합을 return하도록 soltuion 함수를 완성해주세요.
2. 정수 num1과 num2가 주어질 때, num1에서 num2를 뺀 값을 return하도록 soltuion 함수를 완성해주세요.
3. 정수 num1, num2가 매개변수 주어집니다. num1과 num2를 곱한 값을 return 하도록 solution 함수를 완성해주세요.
4. 정수 num1, num2가 매개변수로 주어질 때, num1을 num2로 나눈 몫을 return 하도록 solution 함수를 완성해주세요.
5. 정수 num1, num2가 매개변수로 주어질 때, num1를 num2로 나눈 나머지를 return 하도록 solution 함수를 완성해주세요.

# 1
def solution(num1, num2):
    return num1+num2
# 2
def solution(num1, num2):
    return num1-num2
# 3
def solution(num1, num2):
    return num1*num2
# 4
def solution(num1, num2):
    return num1//num2
# 5
def solution(num1, num2):
    return num1%num2

 

문제 6. 두 수의 나눗셈

프로그래머스

 

정수 num1과 num2가 매개변수로 주어질 때, num1을 num2로 나눈 값에 1,000을 곱한 후 정수 부분을 return 하도록 soltuion 함수를 완성해주세요.

def solution(num1, num2):
    return int (num1/num2 * 1000)

 

문제 7. 숫자 비교하기

프로그래머스

정수 num1과 num2가 매개변수로 주어집니다. 두 수가 같으면 1 다르면 -1을 retrun하도록 solution 함수를 완성해주세요.

def solution(num1, num2):
    if num1 == num2:
        return 1
    else:
        return -1

 

더 간단한 풀이

 

def solution(num1, num2):
    return 1 if num1==num2 else -1

 

문제 8. 분수의 덧셈

프로그래머스

첫 번째 분수의 분자와 분모를 뜻하는 numer1, denom1, 두 번째 분수의 분자와 분모를 뜻하는 numer2, denom2가 매개변수로 주어집니다. 두 분수를 더한 값을 기약 분수로 나타냈을 때 분자와 분모를 순서대로 담은 배열을 return 하도록 solution 함수를 완성해보세요.

def solution(numer1, denom1, numer2, denom2):
    a = numer1*denom2 + numer2*denom1
    b = denom1 * denom2
    
    n = min(a,b)
    gcd = 1
    for i in range(1, n+1):
        if a % i == 0 and b % i == 0:
            gcd = i
            
	return [a/gcd, b/gcd]

 

math.gcd 이용

- gcd( ) : 최대공약수

- lcm( ) : 최소공배수

import math

def solution(denum1, num1, denum2, num2):
    denum = denum1 * num2 + denum2 * num1
    num = num1 * num2
    gcd = math.gcd(denum, num)
    return [denum//gcd, num//gcd]

 

문제 9. 배열 두 배 만들기

프로그래머스

정수 배열 numbers가 매개변수로 주어집니다. numbers의 각 원소에 두배한 원소를 가진 배열을 return하도록 solution 함수를 완성해주세요.

# 1
def solution(numbers):
    answer = []
    for i in numbers:
        a = i*2
        answer.append(a)
    return answer
    
# 2
def solution(numbers):
    return [i*2 for i in numbers]

 

문제 10. 각도기

프로그래머스

각에서 0도 초과 90도 미만은 예각, 90도는 직각, 90도 초과 180도 미만은 둔각 180도는 평각으로 분류합니다. 각 angle이 매개변수로 주어질 때 예각일 때 1, 직각일 때 2, 둔각일 때 3, 평각일 때 4를 return하도록 solution 함수를 완성해주세요.

def solution(angle):
    if angle < 90:
        return 1
    elif angle == 90:
        return 2
    elif angle < 180:
        return 3
    else:
        return 4

 

멋진 풀이

def solution(angle):
    answer = (angle // 90) * 2 + (angle % 90 > 0) * 1
    return answer

 

 

마음에 드는 풀이

def solution(angle):
    if angle<=90:
        return 1 if angle<90 else 2
    else:
        return 3 if angle<180 else 4