백준 1292번 쉽게푸는문제

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

수열의 값이 나오는 함수를 구현한 후 풀어줬습니다.

let input = readLine()!
let array = input.split(separator: " ").map{String($0)}
let first = Int(array.first!)!
let last = Int(array.last!)!

func getNum(num: Int) -> Int {
    var count = 1
    var sum = 1
    while num > sum {
        count+=1
        sum+=count
    }
    return count
}
var result = 0
for i in first...last {
    result+=getNum(num: i)
}
print(result)