[백준 10539] 수빈이와 수열 (Swift)

[백준 10539] 수빈이와 수열

10539번: 수빈이와 수열

입력으로 주어지는 수열이 수열 A가 아닌, 수열 B라는 것 이 포인트이다.

평균 식을 그대로 쓰면 안되고, 평균 식을 변형해야 한다.

수열 B의 길이가 4이고 첫 요소가 3이라면, 수열 A의 첫 요소도 3이다. (첫 요소는 1로 나누므로 값이 같음)

예를 들어 수열 B의 두 번째 요소가 2라면, 이 값은 (수열 A의 첫 요소 + 수열 A의 두 번째 요소) / 2 와 같아야 한다.

이때 수열 A의 첫 번째 요소가 3이므로, 수열 A의 두 번째 요소는 2 * 2 - 3 = 1이 된다.

따라서 수열 A의 n번째 요소 = (수열 B의 n번째 요소 * n) - 수열 A의 n-1 번째 요소까지의 총합 이 된다.

풀이

let count = Int(readLine()!)!
let inputArr = readLine()!.split(separator: " ").map { Int($0)! }
var sum = 0
var resultArr: [Int] = []
(1...count).forEach { num in
    let result = inputArr[num - 1] * num - sum
    sum += result
    print(result)
}