[Python] 백준 #1744 수 묶기

www.acmicpc.net/problem/1744

1744번: 수 묶기

result=0

n=int(input())
data=[]
for i in range(n):
    data.append(int(input()))

#음수 따로,양수따로 리스트 분리 후, 양수는 내림차순, 음수는 오름 차순으로..

m_data=[i for i in data if(i<=0)]
m_data=sorted(m_data) #오름차순

p_data=[i for i in data if(i>0)]
p_data=sorted(p_data,reverse=True)


#마이너스,0 리스트 부터
if(len(m_data)%2==0): # 데이터 개수가 짝수일때
    for i in range(len(m_data)-1):
        if(i%2==0):
            result=result+(m_data[i]*m_data[i+1])        
        
else:
    for i in range(len(m_data)-1):
        if(i%2==0):
            result=result+(m_data[i]*m_data[i+1])
    result+=m_data[len(m_data)-1]


#양수 리스트

if(len(p_data)%2==0): # 데이터 개수가 짝수일때
    for i in range(len(p_data)-1):
        if(i%2==0):
            if(p_data[i+1]==1): #1이면 더하는게 더 큼
                result=result+(p_data[i]+p_data[i+1])
            else:
                result=result+(p_data[i]*p_data[i+1])        
        
else:
    for i in range(len(p_data)-1):
        if(i%2==0):
            if(p_data[i+1]==1): #1이면 더하는게 더 큼
                result=result+(p_data[i]+p_data[i+1])
            else:
                result=result+(p_data[i]*p_data[i+1])
    result+=p_data[len(p_data)-1]
    

print(result)