본문 바로가기

Algorithm/Python

015 - 수들의 합

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