백준 25634번 전구 상태 뒤집기 (JAVA)

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

25634번: 전구 상태 뒤집기

 

저는 구현을 통해 문제를 해결했습니다.

앞에서부터 뒤집었을 때 합을 구하였습니다. 이때 음수라면 0으로 다시 초기화를 하고 아니라면 이 값들의 최대를 구하였습니다.

마지막으로 최대값이 0이라는 것은 모두 켜져있다고 판단하여 가장 작은 전구 값을 빼서 답을 구하고, 아니라면 켜져있는 전구의 합에서 max를 더해주어 답을 구하였습니다. 

 

package BOJ.etc;

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.StringTokenizer;
import java.util.function.Function;

public class BOJ_25634 {
    private static final int INF = 200_000 + 1;
    private static final int ON = 1;
    private static final int OFF = 0;
    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        Function<String,Integer> stoi = Integer::parseInt;
        int n = stoi.apply(br.readLine());
        StringTokenizer st = new StringTokenizer(br.readLine());
        int[] input = new int[n];
        int[] onOff = new int[n];
        int min = INF;

        for(int i = 0 ; i < n ; i++){
            input[i] = stoi.apply(st.nextToken());
            min = Math.min(min,input[i]);
        }
        st = new StringTokenizer(br.readLine());
        int result = 0;
        for(int i = 0 ; i < n ; i++){
            onOff[i] = stoi.apply(st.nextToken());
            if(onOff[i] == ON){
                result += input[i];
            }
        }
        int sum = 0;
        int max = 0;
        for(int i = 0 ; i < n ; i++){
            if(onOff[i] == OFF){
                sum += input[i];
            }
            if(onOff[i] == ON){
                sum -= input[i];
            }
            if(sum < 0){
                sum = 0;
            }
            max = Math.max(sum,max);
        }
        if(max == 0){
            System.out.println(result - min);
        }else{
            System.out.println(result + max);
        }
    }
}