[백준] 2587번 대표값2(with python)

https://www.acmicpc.net/problem/2587

2587번: 대표값2

 

1. 내가 작성한 코드

arr = []
avg = 0
mid = 0

for i in range(5):
  arr.append(int(input()))

arr = sorted(arr)

avg = int(sum(arr)/5)
mid = arr[5//2]

print(avg)
print(mid)

길이가 5로 주어졌기 때문에 mid = arr[2]라고 적어도 되는 거였네 ㅇㅅㅇ..

 

2. 고칠 점

리스트 arr = [ ] 를 선언하고 다시 append 할 필요 없이 arr = [int(input()) for _ in range(5)] 코드를 이용하면 된다고 한다. 

 

mid, avg 변수를 선언하고 print 할 필요 없이 print 함수 안에 int(sum(arr)/5), arr[2]를 각각 넣으면 된다. 

 

3. 수정 후 코드

arr = [int(input()) for _ in range(5)]

arr = sorted(arr)

print(int(sum(arr)/5))
print(arr[2])