문제
문제 링크
n개의 정수로 이루어진 임의의 수열이 주어진다. 우리는 이 중 연속된 몇 개의 수를 선택해서 구할 수 있는 합 중 가장 큰 합을 구하려고 한다. 단, 수는 한 개 이상 선택해야 한다.
예를 들어서 10, -4, 3, 1, 5, 6, -35, 12, 21, -1 이라는 수열이 주어졌다고 하자. 여기서 정답은 12+21인 33이 정답이 된다.
입력
첫째 줄에 정수 n(1 ≤ n ≤ 100,000)이 주어지고 둘째 줄에는 n개의 정수로 이루어진 수열이 주어진다. 수는 -1,000보다 크거나 같고, 1,000보다 작거나 같은 정수이다.
출력
첫째 줄에 답을 출력한다.
Code
import sys
input = sys.stdin.readline
n = int(input().rstrip())
list_ = list(map(int, input().rstrip().split()))
cnt = list_.copy()
for i in range(1,n) :
if cnt[i-1] > 0 :
cnt[i] = cnt[i-1] + list_[i]
else :
pass
print(max(cnt))코드 설명
import sys
input = sys.stdin.readline
n = int(input().rstrip())
list_ = list(map(int, input().rstrip().split()))
cnt = list_.copy()
빠른 입력을 받을 수 있도록 input을 선언하고 테스트 케이스의 개수를 받는다. 이후 list로 들어오는 수를 split해서 받은 후에 최댓값을 카운트할 리스트를 카피한다.
for i in range(1,n) :
if cnt[i-1] > 0 :
cnt[i] = cnt[i-1] + list_[i]
else :
pass
print(max(cnt))앞 수가 음수일 경우 cnt에 저장된 숫자를 그대로 두고, 양수일 경우 i-1번째 수와 i번째 수를 더하였다.
리뷰
그냥 혹시 풀리나? 하고 앞 수가 음수일 경우 cnt에 저장된 값을 그대로 저장하고, 양수면 list에 저장된 값을 더해서 저장하는 풀이로 풀어봤는데 바로 풀렸다. 이 문제 풀 때만 해도 3 -4 5 6 7 8 같은 경우의 수가 있다고 생각했는데 천천히 다시 생각해보니까 3 -4 5 6 7 8을 모두 더하는 것보다 5 6 7 8을 더하는 게 더 크다는 사실을 알아냈다.
