프로그래머스 · 숫자 비교하기

문제

정수 num1, num2가 주어진다(둘 다 0 이상 10,000 이하). 두 수가 같으면 1, 다르면 -1을 반환한다. 예를 들어 2와 3은 -1, 11과 11은 1이다.

풀이

num1과 num2가 같으면 1, 아니면 -1을 반환하는 조건 표현식으로 한 줄에 썼다. 한 문장으로 쓰는 편이 읽기에도 낫다고 생각하지만, 지나치게 줄이거나 늘릴 필요는 없다.

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

다른 풀이

원글에는 if 문으로 풀어 쓴 코드가 함께 있었다. 다만 이 코드는 다른 경우에 -1이 아니라 0을 반환하므로, 같은 결과를 내려면 else 쪽을 answer = -1로 고쳐야 한다.

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

출처

• 프로그래머스 · 숫자 비교하기 (opens in a new tab)