P.1912 연속합

8646 단어 CodingTestCodingTest

1912 연속합

시간 제한메모리 제한제출정답맞힌 사람정답 비율
1 초 (추가 시간 없음) 128 MB98474337842346433.037%

문제

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

코드

import java.io.*;
import java.util.Arrays;

public class P_1912 {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

        int n = Integer.parseInt(br.readLine());
        int[] dp = Arrays.stream(br.readLine().split(" ")).mapToInt(Integer::parseInt).toArray();
        int max = dp[0];

        for (int i = 1; i < n; i++) {
            if (dp[i - 1] > 0 && dp[i - 1] + dp[i] > 0) dp[i] = dp[i - 1] + dp[i];
            max = (max > dp[i]) ? max : dp[i];
        }

        bw.write(Integer.toString(max));
        bw.flush();
    }
}

코드 내용

이전 숫자를 더해가면서 dp를 업데이트 해 나가는 방식을 이용했다.
하지만, 더했을 때 음수라면 최댓값을 절대로 구할 수 없기 때문에 이 때의 dp는 구하지 않았다.

첫제출에서 틀렸는데 dp[i - 1]이 양수인지 음수인지를 고려하지 않았기 때문이다.
dp[i - 1]가 -4이고 dp[i]가 5일 때, 더하면 1로 양수가 나와서 dp가 없데이트 되지만 dp[i - 1] + dp[i]보다 dp[i]만을 이용하는게 더 값이 크기 때문에 이 때는 더해서는 안된다.
그래서 dp[i - 1]이 양수일 때만 dp를 업데이트하도록 하는 조건을 추가했다.

좋은 웹페이지 즐겨찾기