백준 - 수 묶기(Swift)

백준 - 수 묶기(Swift)

 

 

 문제 설명

 

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

1744번: 수 묶기

 

 나의 풀이

 

처음에는 스택을 정의해서 1보다 큰 값이 2개가 들어온다면 곱해주고 아니라면 더해줘서 문제를 풀었더니 음수 두개가 곱해진 경우는 양수가 된다는 사실을 간과하고 문제를 해결하지 못했다. 

 

이 문제는 1보다 큰 수, 1인 수, 0보다 작거나 같은 수 3개로 분류해서 풀어야한다.

1보다 큰수는 전부 곱해주고, 1인수는 전부 더해주며 0보다 작거나 같은 수도 전부 곱해주면된다.

단 1보다 큰수도 그렇고 0보다 작거나 같은 수의 개수가 홀수개라면 곱했을때 가장 작은 값을 만드는 숫자만 더해주고 짝수인채로 곱해줘야한다.

 

문제에 대해 분류를 확실하게 해서 규칙을 찾는 연습이 필요하다.

 

import Foundation

let n = Int(readLine()!)!
var plus: [Int] = []
var minus: [Int] = []
var sum = 0
for _ in 0..<n {
    let a = Int(readLine()!)!
    if a <= 0 { minus.append(a) }
    else if a == 1 { sum += 1 }
    else { plus.append(a) }
}

plus.sort(by: >)
minus.sort(by: <)

if plus.count % 2 == 1 { sum += plus.removeLast()}
if minus.count % 2 == 1 { sum += minus.removeLast()}

while plus.count > 0 || minus.count > 0 {
    if plus.count > 0 { sum += plus.removeLast() * plus.removeLast() }
    if minus.count > 0 { sum += minus.removeLast() * minus.removeLast() }
}

print(sum)