[Kotlin/프로그래머스] 코딩테스트 연습 > 연습문제 > 연속된 부분 수열의 합

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
class Solution {
    fun solution(sequence: IntArray, k: Int): IntArray {
        var answer: IntArray = intArrayOf(0, Int.MAX_VALUE-1)
        var (l, r) = 0 to 0
        var sum = sequence[0]
        while (l <= r && r in 0..sequence.lastIndex) {
            if (sum > k) {
                sum -= sequence[l]
                l++
            } else if (sum < k) {
                if (r < sequence.lastIndex) {
                    r++
                    sum += sequence[r]
                } else {
                    break
                }
            } else {
                if (answer[1] - answer[0] + 1 > r - l + 1) {
                    answer[0] = l
                    answer[1] = r
                }
                sum -= sequence[l]
                l++
            }
        }
        return answer
    }
}