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

1. 괄호(백준 9012번)

 

9012번: 괄호 (acmicpc.net)

9012번: 괄호

 

▷ 풀이 코드

import java.awt.desktop.SystemSleepEvent;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Scanner;

public class GwalHo {
	
	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());
		
		
		for(int i=0; i<n; i++) {
			String input = br.readLine();
						
			int a = 0;
			int b = 0;
			
			for(int j=0; j < input.length(); j++) {
				if(input.charAt(j) == '(') {
					a++;
				} else {
					b++;
				}
				
				if(a<b) break;
			}
			
			if(a == b) {
				sb.append("YES").append("\n");
			} else {
				sb.append("NO").append("\n");
			}
		}
		System.out.println(sb);
	}
	
}

◎ 스택 함수를 이용했을 때

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

public class GwalHo2 {
	
	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());
		
		
		for(int i=0; i<n; i++) {
			String input = br.readLine();
			
			Stack<Character> s = new Stack<>();
						
			int a = 1;
			
			for(int j=0; j < input.length(); j++) {
				if(input.charAt(j) == '(') {
					s.push(input.charAt(j));
					
				} else {
					
					if(s.empty()) {
						a = 0;
						break;
					}
					s.pop();		

				}
			}					
			
			if(a == 0 || s.size() > 0) {
				sb.append("NO").append("\n");
			} else {
				sb.append("YES").append("\n");
			}
		}
		System.out.println(sb);
	}
}

 

 

2. 제로(백준 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);
	}
}

 

 

3. 스택(백준 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();
	}
}

◎ 스택 함수를 이용했을 때

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

public class StackEx {
	
	public static void main(String args[]) throws IOException {
		
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringBuilder sb = new StringBuilder();
				
		Stack<Integer> stack = new Stack<>(); 
		
		int num = Integer.parseInt(br.readLine());
		
			
		for(int i=0; i<num; i++) {
			String behav[] = br.readLine().split(" ");
			
			switch (behav[0]) {
			case "push": 
				stack.add(Integer.parseInt(behav[1]));
				break;
			case "pop" : 
				sb.append(stack.isEmpty()?-1:stack.pop()).append("\n");
				break;
			case "size" : sb.append(stack.size()).append("\n");
				break;
			case "empty" : sb.append(stack.isEmpty()?1:0).append("\n");
				break;
			case "top" : sb.append(stack.isEmpty()?-1:stack.peek()).append("\n");
				break;
			}
		}
		System.out.println(sb);

	}
}

 

 

이번에는 알고리즘 스터디를 통해 배운 스택을 통해 백준 문제 풀이를 해보았습니다!!

 

스택의 FIFO 구조를 이용하면 좀 더 간편하게 값을 저장하고 출력할 수 있어요!!

 

많은 분들의 피드백은 언제나 환영합니다!  많은 댓글 부탁드려요~~