DynamicProgramming1_13연속합 (1912)

문제

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

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

입력

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

출력

첫째 줄에 답을 출력한다.

풀이

• 큐를 이용할 것이다.
• while 문의 array값이 false가 될 때까지
• pop 과 append 반복하다가 k 번째에 새로운 배열에 append 시킬 것이다

코드

내가 생각한 코드-> 시간초과

import sys
sys.stdin = open ("input.txt", "rt")

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

n = int(input())
arr = list(map(int,input().split()))
dp=[0]*(n)
dp[0] = max(arr[0],0)
stdNum = 0

for i in range(n):
    sum = 0
    for j in range(stdNum,i):
        sum += arr[j]
        if sum < 0:
            sum = 0
            stdNum = j+1
        dp[i] = max(dp[i-1],sum)
print(max(dp))

올바른 코드

import sys
sys.stdin = open ("input.txt", "rt")

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

n = int(input())
arr = list(map(int,input().split()))
dp = [0]*(n)
dp[0] =  arr[0]

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

배운 것

dp[i] = max(arr[i], dp[i-1]+arr[i])

코멘트

간단하네..