코딩테스트 대비/단계별 코딩 테스트 준비(27일 과정)

[투 포인터/Python] 1806번: 부분합 - 효과는 굉장했다!

bluetag_boy 2022. 2. 19. 17:23
반응형
 

1806번: 부분합

첫째 줄에 N (10 ≤ N < 100,000)과 S (0 < S ≤ 100,000,000)가 주어진다. 둘째 줄에는 수열이 주어진다. 수열의 각 원소는 공백으로 구분되어져 있으며, 10,000이하의 자연수이다.

www.acmicpc.net

 

알고리즘 분류

  • 두 포인터

 

 

SOLUTION

import sys

N, S = map(int, sys.stdin.readline().split())
nums = list(map(int, sys.stdin.readline().split()))

left, right = 0, 1
tmp = nums[0]
length = sys.maxsize

while True:
    if tmp >= S:
        length = min(length, right - left)
        tmp -= nums[left]
        left += 1

    elif right == N:
        break
        
    else:
        tmp += nums[right]
        right += 1
    
if length == sys.maxsize:
    print(0)

else:
    print(length)