[python/백준] 17744번: 수묶기

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

1744번: 수 묶기

 

❗ 문제

 

어떤 수열이 {0, 1, 2, 4, 3, 5}일 때, 그냥 이 수열의 합을 구하면 0+1+2+4+3+5 = 15이다. 하지만, 2와 3을 묶고, 4와 5를 묶게 되면, 0+1+(2*3)+(4*5) = 27이 되어 최대가 된다.

수열의 모든 수는 단 한번만 묶거나, 아니면 묶지 않아야한다.

 

 

 

❗코드분석

 

 

양수, 음수로 나누어서 리스트를 만들고 하나씩 뺴서 곱한후에 답 출력

 

 

 

 

 

 

❗코드

 

import sys

input = sys.stdin.readline
n = int(input())
plus = []
minus = []
res = 0
for i in range(n):
    a = int(input())
    if a == 1:
        res += 1
    elif a > 0:
        plus.append(a)
    else:
        minus.append(a)
plus.sort()
minus.sort(reverse=True)
while len(plus) != 0:
    if len(plus) == 1:
        res += plus.pop()
    else:
        res += plus.pop() * plus.pop()
while len(minus) != 0:
    if len(minus) == 1:
        res += minus.pop()
    else:
        res += minus.pop() * minus.pop()
print(res)