[프로그래머스] 연속된 부분 수열의 합

https://school.programmers.co.kr/learn/courses/30/lessons/178870

프로그래머스

 

접근

---

• 길이가 최소이며, 앞쪽에 등장하는 수열 찾기 → 투포인터

 

로직

---

• 구간 누적 합이 k보다 작거나 같을 때, end 포인터를 갱신해야 한다. 이때 end 포인터가 마지막 인덱스를 가리키고 있다면, 빠른 종료를 시킨다.

 

코드

---

import Foundation

func solution(_ sequence:[Int], _ k:Int) -> [Int] {
    
    var start = 0, end = 0
    var sum = sequence.first!
    var answer = [0, sequence.count]
    while start < sequence.count {
        if sum == k {
            if end-start < answer[1]-answer[0] {
                answer = [start, end]
            }
            if end + 1 < sequence.count {
                end += 1
                sum += sequence[end]
            } else {
                break
            }
        }
        else if sum < k {
            if end + 1 < sequence.count {
                end += 1
                sum += sequence[end]
            } else {
                break
            }
        }
        else {
            sum -= sequence[start]
            start += 1
        }
    }
    
    return answer
}