[알고리즘] Stack 문제 풀이(백준 10773번, 10828번)

1. 제로(백준 10773번 문제)

 

10773번: 제로 (acmicpc.net)

10773번: 제로

 

▷ 풀이 코드

import java.util.Scanner;
import java.util.Stack;

public class StackZero {
	
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		// stack 내장 함수 사용하기
		Stack<Integer> s = new Stack<>();
		
		int k = sc.nextInt();
		int a;
		int sum = 0;
		
		for(int i=0; i<k; i++) {
			do {
				a = sc.nextInt();
				
				if(a == 0) {
					sum -= s.pop();
				} else {
					
					sum += s.push(a);
				}
				
			} while(a > 100000 && a < 1);
		}
				
		System.out.println(sum);
	}
}

 

 

2. 스택(백준 10828번 문제)

 

10828번: 스택 (acmicpc.net)

10828번: 스택

 

▷ 풀이 코드

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

public class Stack {
	
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringBuilder sb = new StringBuilder();
		
		int n = Integer.parseInt(br.readLine());
        
		int[] arr = new int[n];
		int ptr = 0;
		String a;
						
		for(int i=0; i<n; i++) {
			a = br.readLine();
			
			switch(a) {
			case "top" : 
				if(ptr == 0) System.out.println(-1);
				else System.out.println(arr[ptr - 1]);
				break;
			case "pop" : 
				if(ptr == 0) System.out.println(-1);
				else System.out.println(arr[--ptr]);
				break;
			case "size" : System.out.println(ptr);
				break;
			case "empty" : 
				if(ptr == 0) System.out.println(1);
				else System.out.println(0);
				break;
			default :
				if(a.substring(0, 4).equals("push")) {
					arr[ptr++] = Integer.parseInt(a.substring(5));
				}
				break;
			}
		}				
        br.close();
	}
}