백준 2003번 수들의 합 2 (Python 3)

Baekjoon online judge 2003번 파이썬

 

링크

https://www.acmicpc.net/problem/2003

2003번: 수들의 합 2

두 포인터 문제

i==j 인 경우도 카운트 (a[left:right] slicing에서 right==left+1 이면 a[left] 가 된다.)

 

소스코드

import sys

input = sys.stdin.readline

n, m = map(int, input().split())
a = list(map(int, input().split()))

left = 0
right = 1

cnt = 0

while left <= right and right <= n:
    acc = sum(a[left:right])

    if acc == m:
        cnt += 1
        right += 1
    elif acc < m:
        right += 1
    else:
        left += 1

print(cnt)