테페리넷

이 문서는 읽기 전용입니다. 원본을 볼 수는 있지만 바꿀 수는 없습니다. 문제가 있다고 생각하면 관리자에게 문의하세요.

====== 수들의 합 4 ======

===== 풀이 =====
  * A[i]부터 A[j]까지의 부분합은 A[0]부터 A[j]까지의 누적합에서 A[0]부터 A[i-1]까지의 누적합을 뺀 값과 같다. 
  * 따라서 원래 배열에서 부분합이 K인 구간을 찾는 것은, 누적합 배열에서 차이가 K인 두 원소를 찾는것과 동일하다. 이는 간단하게 O(n)에 구현 가능 

===== 코드 =====
<dkpr py>
"""Solution code for "BOJ 2015. 수들의 합 4".

- Problem link: https://www.acmicpc.net/problem/2015
- Solution link: http://www.teferi.net/ps/problems/boj/2015

Tags: [Prefix sum]
"""

import collections


def main():
    N, K = [int(x) for x in input().split()]  # pylint: disable=unused-variable
    A = [int(x) for x in input().split()]

    counter = collections.defaultdict(int, {0: 1})
    answer = 0
    p_sum = 0
    for a_i in A:
        p_sum += a_i
        answer += counter[p_sum - K]
        counter[p_sum] += 1

    print(answer)


if __name__ == '__main__':
    main()
</dkpr>

{{tag>BOJ ps:problems:boj:골드_4}}