본문 바로가기

알고리즘/초급1

[JAVA ] 백준 13398번: 연속합 2 ( 기초 1-32 )

문제

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

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

만약, -35를 제거한다면, 수열은 10, -4, 3, 1, 5, 6, 12, 21, -1이 되고, 여기서 정답은 10-4+3+1+5+6+12+21인 54가 된다.

입력

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

 

import java.util.Arrays;
import java.util.Scanner;

public class baekjoon13398_2 {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int[] info = new int[n];
        int[] forward = new int[n];
        int[] reverse = new int[n];
        int[] total = new int[n];

        for (int i = 0; i < n; i++) {
            info[i] = sc.nextInt();
        }

        forward[0] = info[0];
        for (int i = 1; i < n; i++) {
            forward[i] = info[i];
            if (forward[i - 1] >= 0) {
                forward[i] = forward[i - 1] + forward[i];
                continue;
            }
        }

        reverse[n - 1] = info[n - 1];
        for (int i = n - 2; i >= 0; i--) {
            reverse[i] = info[i];
            if (reverse[i + 1] >= 0) {
                reverse[i] = reverse[i + 1] + reverse[i];
            }
        }

        int res = Arrays.stream(forward).max().getAsInt();

        for (int i = 1; i < n - 1; i++) {
            if (res < forward[i - 1] + reverse[i + 1]) {
                res = forward[i - 1] + reverse[i + 1];
            }
            total[i] = forward[i - 1] + reverse[i + 1];
        }


        System.out.println(res);
    }
}