[BOJ1541] 잃어버린 괄호 (Java)

문제

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

1541번: 잃어버린 괄호

풀이

마이너스 기호가 한 번이라도 나온다면 그 뒤에 나오는 숫자는 모두 음수로 더해질 수 있다. 

5+10-(20+30+50)-(20+30)

 

단 초항은 양수이므로, -가 처음 나오기 전까지의 숫자는 양수로 더해져야 함

5+10-(20+30+50)-(20+30)

5+10 은 양수로 더하고 

나머지 숫자는 음수로 더하여 합계를 내면 된다. 

 

만약 - 기호가 없다면 모두 양수이므로 모든 숫자를 더하면 된다.

'-' 기호로 split하여 첫번째 덩어리와 나머지 덩어리를 분리하여 더한 후 

- 기호 존재 여부에 따라 답을 산출한다.

소스코드

package greedy;

import java.io.BufferedReader;
import java.io.InputStreamReader;

public class BOJ1541 {

    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String s = br.readLine();

        String[] arr = s.split("-");
        String[] subarr = null;
        int sum = 0;
        int first = 0;
        for(int i = 0; i < arr.length; i++){
            subarr = arr[i].split("\\+");
            for(int j = 0;j < subarr.length; j++){
                int num = Integer.parseInt(subarr[j]);
                if(i == 0) {
                    first += num;
                } else {
                    sum += num;
                }
            }
        }

        if(s.contains("-")){
            System.out.println(first-sum);
        } else {
            System.out.println(first+sum);
        }
    }
}