Java 스택, 큐 알고리즘3

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) 알고리즘 문제풀이 : 코딩테스트 대비 - 인프런 | 강의

 

문제 5 : 

출처 - https://www.acmicpc.net/problem/10799

 

 

 

소스코드1 :

import java.util.*;
class Main {
	public int solution(String str) {
		int answer = 0;
		boolean laser = false;
		Stack<Character> stack = new Stack<>();
		for(char x : str.toCharArray()) {
			if (x == '(') {
				stack.push(x);
				laser = true;
			}
			else {
				// 레이저 발사
				if (laser == true) {
					stack.pop();
					answer += stack.size();
					laser = false;
				}
				// 막대기 하나 끊기
				else {
					stack.pop();
					answer++;
				}
			}
		}
		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));
	}
}

그런데 굳이 이렇게 boolean으로 레이저인지 막대기 끊기인지 판별하지 않고

인덱스로 판별하는 방법이 코드가 더 깔끔하다.

 

소스코드2 :

import java.util.*;
class Main {
	public int solution(String str) {
		int answer = 0;
		boolean laser = false;
		Stack<Character> stack = new Stack<>();
		for(int i=0; i<str.length(); i++) {
			if (str.charAt(i) == '(') stack.push('(');
			else {
				stack.pop();
				if (str.charAt(i-1) == '(') answer += stack.size();
				else answer++;
			}
		}
		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));
	}
}