[BOJ: 1912] - Python / 파이썬 - 연속합

o_jooon_·2024년 2월 27일
0

BOJ

목록 보기
33/44
post-thumbnail

서론

DP 문제입니다.
DP를 이용한 연속합의 가장 기초 문제입니다.

난이도

실버 2


문제

조건

시간 제한메모리 제한
1 초 (추가 시간 없음)128 MB

n개의 정수로 이루어진 임의의 수열이 주어진다. 우리는 이 중 연속된 몇 개의 수를 선택해서 구할 수 있는 합 중 가장 큰 합을 구하려고 한다. 단, 수는 한 개 이상 선택해야 한다.

예를 들어서 10, -4, 3, 1, 5, 6, -35, 12, 21, -1 이라는 수열이 주어졌다고 하자. 여기서 정답은 12+21인 33이 정답이 된다.


입력

첫째 줄에 정수 n(1 ≤ n ≤ 100,000)이 주어지고 둘째 줄에는 n개의 정수로 이루어진 수열이 주어진다. 수는 -1,000보다 크거나 같고, 1,000보다 작거나 같은 정수이다.


출력

첫째 줄에 답을 출력한다.


예시

예제 입력 1

10
10 -4 3 1 5 6 -35 12 21 -1

예제 출력 1

33

예제 입력 2

10
2 1 -4 3 4 -4 6 5 -5 1

예제 출력 2

14

예제 입력 3

5
-1 -2 -3 -4 -5

예제 출력 3

-1

풀이

DP를 이용하여 현재 위치(i)의 이전 dp값에 자기 자신을 더한 값(dp[i - 1] + num[i])과 자기 자신(num[i]) 중, 더 큰 값을 선택하여 dp에 넣어주면 됩니다.

코드

import sys
input = sys.stdin.readline

n = int(input())
nums = list(map(int, input().split()))
dp = [nums[0]] + [0] * (n - 1)	# DP 생성

for i in range(1, n):
	# 이전 dp값에 자기 자신을 더한 값과 자기 자신의 값 중 더 큰 값을 선택
    dp[i] = max(dp[i - 1] + nums[i], nums[i])

print(max(dp))	# 가장 큰 값 출력

실행 결과

profile
iOS개발 공부 중입니다.

0개의 댓글