[파이썬 알고리즘] 수들의 합

수들의 합

📄 문제

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을 넘지 않는 자연수이다.

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

▣ 입력예제
8 3 1 2 1 3 1 1 1 2

▣ 출력예제
5

🧞 풀이

풀이 (시간 초과)

# 입력: 첫째 줄 - N, M, 둘째 줄 - 수열 나열
# 출력: 합이 M이 되는 경우의 수 출력

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

for i in range(n):
    for j in range(i+1, n+1):
        if sum(list_n[i:j]) == m:
            a += 1

print(a)

 

다른 사람 풀이

# 연속적인 부분 수열의 합이 m과 같은 경우 출력
# 포인터 각각 할당 lt, rt

n, m = map(int, input().split())
a = list(map(int, input().split()))
lt = 0                              # 포인터 lt 할당
rt = 1                              # 포인터 rt 할당
tot = a[0]                          # tot 초기값: a[0]
cnt = 0                             # 개수 세기 위한 cnt 할당

while True:                         # break 되기 전까지 반복
    if tot < m:                     # 합이 m보다 작을 경우
        if rt < n:                  # rt가 n보다 작을 경우
            tot += a[rt]            # tot에 a[rt] 값 더해줌
            rt += 1                 # rt 포인터 1만큼 증가
        else:                       # rt가 n보다 크거나 같으면
            break                   # 반복문 종료

    elif tot == m:                  # 만약 합이 m과 같으면
        cnt += 1                    # cnt 1 증가
        tot -= a[lt]                # tot에서 a[lt] 값 빼줌
        lt += 1                     # lt 1 증가

    else:                           # 만약 합이 m보다 크면
        tot -= a[lt]                # cnt 증가 없이 a[lt] 값 빼줌
        lt += 1                     # lt 1 증가

print(cnt)                          # cnt 출력