[백준] 투 포인터 - 1806번 : 부분합 (Python)

Daisy 🌼·2022년 8월 2일
0

백준

목록 보기
7/7
post-thumbnail

1. 문제 👻

  • 문제
    10,000 이하의 자연수로 이루어진 길이 N짜리 수열이 주어진다. 이 수열에서 연속된 수들의 부분합 중에 그 합이 S 이상이 되는 것 중, 가장 짧은 것의 길이를 구하는 프로그램을 작성하시오.

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

  • 출력
    첫째 줄에 구하고자 하는 최소의 길이를 출력한다. 만일 그러한 합을 만드는 것이 불가능하다면 0을 출력하면 된다.


2. Cording 🥲

문제풀이 : sys.maxsize() 활용

# 이 수열에서 연속된 수들의 부분합 중에 그 합이 S 이상이 되는 것 중, 가장 짧은 것의 길이
#mx = -sys.maxsize / mn = sys.maxsize
# 위와 같이 sys의 모듈인 maxsize를 사용하면 시스템이 지정할 수 있는
# 최댓값과 최소값을 활용할 수 있다.
import sys

N, S = map(int, input().split()) # 10 15
numbers = list(map(int, input().split()))
# [5, 1, 3, 5, 10, 7, 4, 9, 2, 8]

left, right = 0, 0 # 두 개의 포인터는 0에서 부터 시작
sum = 0 # 합을 저장할 변수
min_length = sys.maxsize # 먼저 최대 길이로 지정

while True:
    # 만약 합이 S가 넘는다면,
    # left를 하나씩 옮겨보면서 어디까지 길이가 줄어드나 확인하고
    # 가장 작은 길이를 출력한다.
    if sum >= S: # 5 + 1 + 3 + 5 +10 = 24
        min_length = min(min_length, right - left)
        sum -= numbers[left]
        left += 1
    elif right == N: # right 좌표가 N 길이가 되면 break
        break
    # 만약 총합이 S를 넘지 않는다면,
    # right 을 오른쪽으로 한칸씩 옮기며 총합이 S를 넘을때까지 더함
    # # s와 같아지기 전까지 numbers[0,1,2,...,n] 더해줄 것
    else:
        sum += numbers[right] # 5 + 1 + 3 + 5 = 14
        right += 1

if min_length == sys.maxsize:
    print(0)
else:
    print(min_length)

if와 else를 바꾸는 것이 직관적으로 이해가 잘 되는 것 같아, 바꿔서 제출했더니 런타임 에러가 떴다.

profile
세상을 이롭게하는 AI Engineer

0개의 댓글