Java 스택, 큐 알고리즘1

https://www.inflearn.com/course/%EC%9E%90%EB%B0%94-%EC%95%8C%EA%B3%A0%EB%A6%AC%EC%A6%98-%EB%AC%B8%EC%A0%9C%ED%92%80%EC%9D%B4-%EC%BD%94%ED%85%8C%EB%8C%80%EB%B9%84/dashboard

자바(Java) 알고리즘 문제풀이 : 코딩테스트 대비 - 인프런 | 강의

 

자바 Stack

 

스택 사용하기 위해선 import java.util.Stack;

 

Stack 선언

Stack<Integer> stack = new Stack<>();

 

Stack 값 추가

stack.push(1);
stack.push(2);
// 1이 들어가고 그 위에 2가 들어간다.

 

Stack 값 삭제

stack.pop(); // 스택 가장 위쪽의 값 제거하고 반환
stack.clear(); // 스택의 전체 값 제거

 

Stack 출력 및 기타 메서드

stack.peek(); // 스택 가장 상단의 값 출력
stack.size();
stack.isEmpty(); // 스택이 비어있다면 true, 아니면 false
stack.contains(2); // 스택에 2가 있다면 true, 아니면 false

stack.search(2); // 스택에 2가 있는 위치(인덱스) 반환

 

TIP : pop할 때 isEmpty()로 체크하는 것이 좋다.

 

 

문제 1 :

괄호가 쌍이 맞게 올바로 입력되면 "YES", 아니면 "NO"를 입력하여라

(())()는 올바로 입력된 예시이고 (()()))는 올바르게 입력되지 않은 예시이다.

첫 번째 줄에 괄호 문자열(최대 길이 30)이 입력된다.

입력예제

(()(()))(()

출력예제

NO

 

소스코드 :

import java.util.*;
class Main {
	public String solution(String str) {
		String answer = "YES";
		Stack<Character> stack = new Stack<>();
		for(char x : str.toCharArray()) {
			if (x == '(') stack.push(x);
			else {
				if (stack.isEmpty()) return "NO";
				else stack.pop();
			}
		}
		if (stack.isEmpty() == false) return "NO";

		return answer;
	}

	public static void main(String[] args) {
		Main T = new Main();
		Scanner sc = new Scanner(System.in);
		String str = sc.next();

		System.out.println(T.solution(str));
	}
}

 

 

문제 2 :

입력된 문자열에서 소괄호 사이 존재하는 모든 문자 제거 후 남은 문자만을 출력하는 프로그램을 작성하시오.

문자열의 길이는 100 넘지 않는다.

입력예제

(A(BC)D)EF(G(H)(IJ)K)LM(N)

출력예제

EFLM

 

소스코드 :

import java.util.*;
class Main {
	public StringBuilder solution(String str) {
		StringBuilder answer = new StringBuilder("");
		Stack<Character> stack = new Stack<>();
		for(char x : str.toCharArray()) {
			// ')' 만나면 최상단이 '(' 가 될 때 까지 pop()한다.
			// 최종적으로 '('를 꺼내고 멈추게 된다.
			if (x == ')') {
				while (stack.pop() != '(');
			}
			else stack.push(x);
		}
		// 스택 결과 answer에 담아서 출력하기
		for(int i=0; i<stack.size(); i++) {
			answer.append(stack.get(i));
		}
		return answer;
	}

	public static void main(String[] args) {
		Main T = new Main();
		Scanner sc = new Scanner(System.in);
		String str = sc.next();

		System.out.println(T.solution(str));
	}
}