2003 연속적인 값 구하기

코린이서현이·2024년 3월 12일
0

문제


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

4 2
1 1 1 1

예제 출력 1

3

내 정답 코드

x = [int(i) for i in input().split(" ")]

a_list=[int(i) for i in input().split(" ")]

start = 0
end = 0
result = 0
intervals = 0

while start < len(a_list) and end < len(a_list):
    intervals = sum(a_list[start:end+1])
    if intervals == x[1]:
        result += 1
        if start == end:
            end += 1
        else:
            start += 1
    elif intervals > x[1]:
        start += 1
    else:
        end += 1

print(result)
profile
24년도까지 프로젝트 두개를 마치고 25년에는 개발 팀장을 할 수 있는 실력이 되자!

0개의 댓글