문제
X대학 M교수님은 프로그래밍 수업을 맡고 있다. 교실엔 학생이 30명이 있는데, 학생 명부엔 각 학생별로 1번부터 30번까지 출석번호가 붙어 있다.
교수님이 내준 특별과제를 28명이 제출했는데, 그 중에서 제출 안 한 학생 2명의 출석번호를 구하는 프로그램을 작성하시오.
입력
입력은 총 28줄로 각 제출자(학생)의 출석번호 n(1 ≤ n ≤ 30)가 한 줄에 하나씩 주어진다. 출석번호에 중복은 없다.
출력
출력은 2줄이다. 1번째 줄엔 제출하지 않은 학생의 출석번호 중 가장 작은 것을 출력하고, 2번째 줄에선 그 다음 출석번호를 출력한다.
코드
arr = list(range(1, 31))
for _ in range(28):
num = int(input())
arr.remove(num)
print(arr[0])
print(arr[1])풀이
num = [i+1 for i in range(30)]
lst1 = [int(input()) for j in range(28)]
lst2 = [k for k in num if k not in lst1]
print(min(lst2))
print(max(lst2))나는 위와 같이 코드를 작성하였다. 리스트 num에 있는 숫자가 새로 받은 n에 포함되지 않으면 새로운 리스트에 저장하여 최솟값과 최댓값을 구하는 방식이다. 틀린 답은 아니나 for문을 더 줄일 수 있는 방법이 있지 않을까 싶어서 찾아봤다.
자연수 n을 받을 때마다 1부터 30까지 들어있는 리스트에 숫자 n을 remove하는 방식이다. 이 경우 arr 리스트는 처음부터 오름차순으로 만들어져있으므로 min, max 함수를 쓸 필요가 없게 된다.
