본문 바로가기

파이썬 알고리즘 풀이

15) 수들의 합

문제

N개의 수로 된 수열 A[1], A[2], …, A[N] 이 있다. 이 수열의 i번째 수부터 j번째 수까지의 합 A[i]+A[i+1]+…+A[j-1]+A[j]가 M이 되는 경우의 수를 구하는 프로그램을 작성하기

 

조건

 

제한사항

첫째 줄에 N(1≤N≤10,000), M(1≤M≤300,000,000)이 주어진다.

다음 줄에는 A[1], A[2], …, A[N]이 공백으로 분리되어 주어진다. 각각의 A[x]는 30,000을 넘지 않는 자연수이다.

첫째 줄에 경우의 수를 출력한다.

 

문제풀이

N, M=map(int, input().split())
a=list(map(int, input().split()))

lt=0
rt=1
tot=a[0]
cnt=0

while True:
  if tot<M:
    if rt<N:
      tot+=a[rt]
      rt+=1
    else:
      break
  elif tot==M:
    cnt+=1
    tot-=a[lt]
    lt+=1
  else:
    tot-=a[lt]
    lt+=1

print(cnt)

 

'파이썬 알고리즘 풀이' 카테고리의 다른 글

17) 사과나무(다이아몬드)  (0) 2021.10.22
16) 격자판 최대합  (0) 2021.10.22
14) 두 리스트 합치기  (0) 2021.10.20
13) 카드 역배치(정올 기출)  (0) 2021.10.20
12) 숫자의 추출  (0) 2021.10.20