코딩 테스트(Python)/백준, 프로그래머스
18310 안테나
5_hyun
2022. 1. 27. 21:25
반응형
https://www.acmicpc.net/problem/18310
18310번: 안테나
첫째 줄에 집의 수 N이 자연수로 주어진다. (1≤N≤200,000) 둘째 줄에 N채의 집에 위치가 공백을 기준으로 구분되어 1이상 100,000이하의 자연수로 주어진다.
www.acmicpc.net
내 풀이(맞음)
n = int(input())
ary = list(map(int, input().split()))
m = len(ary)//2 - 1
l = m + 1
ary.sort()
sum1 = 0
sum2 = 0
for i in range(n):
if i == m:
continue
sum1 += abs(ary[m] - ary[i])
for i in range(n):
if i == l:
continue
sum2 += abs(ary[l] - ary[i])
print(ary[m if sum1 <= sum2 else l])
집 위치의 중앙값을 2로 나눠서 m으로 구하고 그 값에서 +1된 자리 l까지 생각을 한다. 그리고 m과 l로 값을 구하고 비교하여 값을 출력하면된다.
반응형