김경록의 앱 개발 여정

[Swift 백준] 2559 수열 본문

Algorithm

[Swift 백준] 2559 수열

Kim Roks 2025. 1. 22. 16:50

풀이 아이디어

  • 슬라이딩 윈도우 알고리즘을 활용합니다

자세한 풀이 방식은 주석을 참고해주세요.

풀이

let nk = readLine()!.split(separator: " ").map{Int($0)!}

// n: 총 날짜수, k: 연속 날짜 수
let (n,k) = (nk[0],nk[1])

let input = readLine()!.split(separator: " ").map{Int($0)!}
//첫 구간합을 구합니다.
var initialSum = input[0..<k].reduce(0, +)
var maxResult = initialSum

for i in k..<n {
    // 윈도우를 움직이며, 가장 왼쪽값(지나간 값)은 빼주고 가장 오른쪽값(새로 들어오는 값)을 더합니다
    initialSum = initialSum - input[i - k] + input[i]
    maxResult = max(maxResult, initialSum)
}

print(maxResult)