[백준] 10773번 문제

문제

문제출처 : Baekjoon

풀이

import java.util.*;

public class Main {
    public int solution(int n, int[] arr){
        Stack<Integer> stack = new Stack<>();
        int sum = 0;

        for(int i = 0; i < n; i++){
            if(arr[i]!= 0){
                stack.push(arr[i]);
            }else {
                stack.pop();
            }
        }

        while(!stack.isEmpty()){
            sum += stack.pop();
        }
        return sum;
    }

    public static void main(String[] args) {
        Main function = new Main();
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int[] arr = new int[n];
        for(int i = 0; i < n; i++){
            arr[i] = sc.nextInt();
        }
        System.out.println(function.solution(n, arr));
    }
}

회고

1. main과 함수를 나누어서 사용하는 방법을 이용하게되어서 arr배열을 따로 선언해줬는데, 배열을 선언하지 않는 방법은 없을까? main에 다 넣어서 계산하는 방법뿐인가? 내 머리로는 여기까지가 한계😶