Python Comprehension 활용 : 백준 1330 [두 수 비교하기]

Hello, I'm Dona!

해당 문제는 사실 엄청 쉬운 문제입니다. 하지만 python에서 자주 사용하는 Comprehension을 연습하기에 아주 좋은 문제라고 생각해서 포스팅하려고 합니다 :)

---

문제 : 두 정수 A와 B가 주어졌을 때, A와 B를 비교하는 프로그램을 작성하시오.

출력 :

첫째 줄에 다음 세 가지 중 하나를 출력한다.

• A가 B보다 큰 경우에는 '>'를 출력한다.
• A가 B보다 작은 경우에는 '<'를 출력한다.
• A와 B가 같은 경우에는 '=='를 출력한다.

  ---

먼저 기본적으로는 아래와 같이 풀 수 있겠죠?!

map 함수는 다른 포스팅에서 다룰 예정인데 알고리즘 문제를 풀 때 매우 유용하니 알고 계시면 정말 정말 좋습니다!

input과 split으로 받은 두 변수를 int형으로 바꾸어주는 효과입니다.

아래처럼 if, elif, else로 나누어서 작성하면 코드가 6줄이라 상당히 길어요!

간단하게 표현하는 방법을 알아보겠습니다!

a, b = map(int, input().split())

if a > b :
    print('>')
elif a < b:
    print('<')
else:
    print('==')
    
print(result)

어떻게 간단하게 표현할 수 있을까요? 아래처럼 result로 한 번에 받아서 표현할 수 있습니다! 

기본적으로는 if와 else로만 표현하는데 else 안에서 한 번 더 if와 else로 나누어서 사용할 수 있습니다 :-)

a, b = map(int, input().split())
result = '>' if a > b else '<' if a < b else '=='
print(result)

---

쉬운 문제로 Python Comprehension 사용하는 방법에 간단히 알아보았습니다. 

오늘 배운 if, else를 활용하는 방법 외의 다른 다양한 Comprehension도 새로운 포스팅으로 소개할게요! 

좋은 하루 보내세요! 🍀

 

문제 출처(백준 알고리즘) : https://www.acmicpc.net/problem/1330