[백준/1919번/파이썬(Python)] 연속합

문제 링크

https://www.acmicpc.net/problem/1912

 

1912번: 연속합

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

www.acmicpc.net

 

문제

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) 문제
  • 배열의 두번째부터 반복문 수행
  • 현재 배열의 값(dp[i])과 현재 배열과 이전까지의 배열의 합(dp[i] + dp[i-1])을 비교해서 더 큰값을 배열에 저장
    ex) dp[1] = -4, dp[1] + dp[0] = -4 + 10 = 6이므로 6을 dp[1]에 저장
  • 해당 반복문을 수행하면 n까지의 최대 값이 dp[n]에 저장됨
  • 저장된 배열 중 최대값 출력

 

소스 코드

n = int(input())
dp = list(map(int, input().split(" ")))

max_sum = dp[0]

for i in range(1, n):
    dp[i] = max(dp[i], dp[i] + dp[i-1])


print(max(dp))

 

댓글