[백준] 1744번 수 묶기 파이썬 코드

코드

 

n = int(input())
positive = [] # 양수 모음
negative = [] # 음수 모음
max_num = 0
for _ in range(n):
  x = int(input())
  if x > 1: positive.append(x) # 1보다 크면 양수 리스트에 넣는다
  elif x == 1: max_num += 1  # 1은 곱하는 것보다 더하는 게 더 큰 수가 나오므로 더해준다
  elif x <1: negative.append(x) # 1보다 작으면 음수 리스트에 넣는다

positive.sort(reverse=True) # (99, 98, 97 ... 1) # 양수는 내림차순으로 정렬
negative.sort() # 음수는 오름차순으로 정렬

if len(positive)%2: # 나머지가 있을 때, 즉 1개가 남을 때 
  for i in range(0, len(positive)-1, 2): # 2개씩 건너뛰면서 더해준다
    max_num += positive[i]*positive[i+1]
  max_num += positive[-1] # 마지막 하나 더해주기
else:
  for i in range(0, len(positive),2): # 개수가 딱 맞아 떨어질 때
    max_num += positive[i]* positive[i+1] 
if len(negative)%2: # 음수리스트 중 1개가 남을 때
  for i in range(0, len(negative)-1, 2):
    max_num += negative[i]*negative[i+1]
  max_num += negative[-1]
else: # 개수가 딱 맞아 떨어질 때
  for i in range(0, len(negative),2):
    max_num += negative[i]*negative[i+1]
print(max_num) # 최대값 출력