28-SeongHoonC

[SeongHoonC]

🔗 문제 링크

부분합

✔️ 소요된 시간

30분

✨ 수도 코드

문제를 보자마자 익숙함이 느껴졌다.
head, tail 와 head 부터 tail 이전까지의 합을 저장하는 sum 을 만든다.

1. sum 이 s 보다 작으면 tail 을 증가시키고 sum 에 arr[tail] 을 더한다.
2. sum 이 s 보다 크거나 같으면
   2-1. 개수가 최소인지 확인하고 최소값으로 답을 업데이트한다.
   2-2. head 를 증가시키고 sum 에 arr[head] 를 뺀다.

위 과정을 head < tail 이고 head, tail 이 배열의 사이즈를 넘기지 않을 때까지 반복한다.

import java.io.BufferedReader
import java.io.InputStreamReader
import kotlin.math.min

private lateinit var br: BufferedReader

fun main() {
    br = BufferedReader(InputStreamReader(System.`in`))
    val (n, s) = br.readLine().split(" ").map { it.toInt() }
    val arr = br.readLine().split(" ").map { it.toInt() }

    var minCount = 100_001
    var head = 0
    var tail = 1
    var sum = arr[head]

    while (true) {
        if (head >= arr.size) {
            break
        }

        // s 보다 크거나 같다면
        if (sum >= s && tail > head) {
            minCount = min(minCount, tail - head)
            sum -= arr[head]
            head++
            continue
        }

        // s 보다 작다면
        if (tail >= arr.size) break
        sum += arr[tail]
        tail++
    }
    println(if (minCount == 100_001) 0 else minCount)
}

📚 새롭게 알게된 내용

[tgyuuAn]

import sys

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

start = 0
end = 0
accumulate = numbers[0]
answer = int(1e9)
while start <= end:
    if accumulate >= S:
        answer = min(answer, end-start+1)
        
        accumulate -= numbers[start]
        start+=1

    elif accumulate < S:
        if end == N-1: break
        else:
            end+=1
            accumulate += numbers[end]

if answer == int(1e9): print(0)
else: print(answer)

이 문제 정말 투 포인터 입문하기 좋은 문제라고 생각됩니다!

문제 추천 게시판에 글 하나 올려주실 수 있나요 ?!

[SeongHoonC]

@tgyuuAn 올리려고 했으나 구교수님이 이미 올리셨군요

[H0ngJu]

처음 이 문제를 보고 푼적이 없는데 어디선가 풀어본 느낌인데? 했더니
저번에 올리셨던 소수의 연속합 문제랑 비슷하네요 ~!

투 포인터 복습했습니다 :)

문제 푸느라 수고하셨습니다 !!

import sys

def input() : return sys.stdin.readline().rstrip()

N, S = map(int, input().split())
arr = list(map(int, input().split()))

start = 0
end = 0
sum = arr[0]
total = int(1e9)

while start<=end:
    if end >= len(arr) or start>= len(arr):
        break
    if sum >= S:
        total = min(total, end-start+1)
        sum -= arr[start]
        start += 1
    else:
        end += 1
        if end < N: sum += arr[end]

if total == int(1e9):
    print(0)
else:
    print(total)