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

프로그래머스

---

풀이

func solution(_ elements:[Int]) -> Int {
    var result = [Int]()
    var count = elements.count
    var arr = Array(repeating: 0, count: count)
    for i in 0..<count-1 {
        for j in 0..<count {
            arr[j] += elements[(j+i)%count]
        }
        result.append(contentsOf: arr)
    }
    return Set(result).count+1
}

 

누적합을 이용해 낮은 시간복잡도로 문제를 풀 수 있었습니다.

 

---