수들의 합 4

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

"""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()