java 6198번 옥상 정원 꾸미기 (백준)

import java.util.Scanner;

class Stack {
private int top = 0;
private int arr[];
public Stack() {top=0; arr = new int[80001];}
public	boolean isEmpty() {return top == 0;}
public	void push(int e) {arr[top++] = e;}
public	void pop() { --top;}
public	int peek() {return arr[top-1];}
public	int cnt() {return top;}
}

public class Main {
	public static void main(String args[]) {
		Stack s = new Stack();
		Scanner scan = new Scanner(System.in);
		int n = scan.nextInt();
		long ans = 0;
		
		while(n > 0) {
			int tmp = 0;
			tmp = scan.nextInt();
			
			if(s.isEmpty() == true) s.push(tmp);
			else {
				if(s.peek() > tmp) s.push(tmp);
				else {
					while(! s.isEmpty() && s.peek() <= tmp) {
						s.pop();
					}
					s.push(tmp);
				}
			}
			ans += s.cnt()-1;
			n--;
		} 
		scan.close();
		System.out.println(ans);
	}
}

 

풀이 : 스택을 class로 구현 했습니다.

 

이 java 코드는 c++코드를 java로 바꾼 것 입니다.

 

https://h202.tistory.com/225

c++ 6198번 옥상 정원 꾸미기 (백준)