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을 넘지 않는 자연수이다.
▣ 출력설명
첫째 줄에 경우의 수를 출력한다.
▣ 입력예제 1
8 3
1 2 1 3 1 1 1 2
▣ 출력예제 1
5
# 효율이 떨어지는 이중 for 문,
# N, M 이 커질수록 2번째 for문에서 처음부터 다시 더히기 때문에 필요없는 시간을 소비한다.
n, m = map(int, input().split())
num = list(map(int, input().split()))
cnt = 0
for i in range(n):
for j in range(i+1, n+1):
tot = sum(num[i:j])
if tot < m:
continue
elif tot == m:
cnt += 1
break
else:
break
print(cnt)
# 필요한 배열만 더하고/빼기를 반복해서 속도 효율이 좋은 코딩 예시
n, m = map(int, input().split())
num = list(map(int, input().split()))
cnt = 0
s = 0
e = 1
tot = num[0]
while True:
if tot < m:
if e < n:
tot += num[e]
e += 1
else:
break
elif tot == m:
cnt += 1
tot -= num[s]
s += 1
else:
tot -= num[s]
s += 1
print(cnt)
'Algorithm > Python' 카테고리의 다른 글
017 - 사과나무 (다이아몬드) 합계 (0) | 2023.08.08 |
---|---|
016 - 격자판 최대합 (0) | 2023.08.07 |
014 - 두 리스트 합치기 (0) | 2023.08.04 |
013 - 카드 역배치 (0) | 2023.08.03 |
012 - 숫자만 추출 (0) | 2023.08.03 |