백준 - ATM(Swift)

백준 - ATM(Swift)

 

 

 문제 설명

 

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

11399번: ATM

 

 

 나의 풀이

 

import Foundation

let n = readLine()!
var input = readLine()!.components(separatedBy:" ")
var arr = input.map{ Int($0)! }.sorted(by: <)
var sum = 0
for i in 0..<arr.count {
    for j in 0...i {
        sum += arr[j]
    }
}
print(sum)

 

 다른 사람의 풀이

 

sum변수를 하나 더 만들어서 for문이 돌때마다 누적시키도록 했다.

for문이 하나만 쓰여서 더 효율적이다.

 

import Foundation

let n = readLine()!
var input = readLine()!.components(separatedBy:" ")
var arr = input.map{ Int($0)! }.sorted(by: <)
var sum = 0
var sum2 = 0
for i in 0..<arr.count {
    sum += arr[i]
    sum2 += sum
}
print(sum2)