본문 바로가기
공부 정리/백준

[백준] 자바 3986 좋은 단어

by 경적필패. 2021. 3. 29.
반응형

문제

이번 계절학기에 심리학 개론을 수강 중인 평석이는 오늘 자정까지 보고서를 제출해야 한다. 보고서 작성이 너무 지루했던 평석이는 노트북에 엎드려서 꾸벅꾸벅 졸다가 제출 마감 1시간 전에 깨고 말았다. 안타깝게도 자는 동안 키보드가 잘못 눌려서 보고서의 모든 글자가 A와 B로 바뀌어 버렸다! 그래서 평석이는 보고서 작성을 때려치우고 보고서에서 '좋은 단어'나 세보기로 마음먹었다.

평석이는 단어 위로 아치형 곡선을 그어 같은 글자끼리(A는 A끼리, B는 B끼리) 쌍을 짓기로 하였다. 만약 선끼리 교차하지 않으면서 각 글자를 정확히 한 개의 다른 위치에 있는 같은 글자와 짝 지을 수 있다면, 그 단어는 '좋은 단어'이다. 평석이가 '좋은 단어' 개수를 세는 것을 도와주자.

입력

첫째 줄에 단어의 수 N이 주어진다. (1 ≤ N ≤ 100)

다음 N개 줄에는 A와 B로만 이루어진 단어가 한 줄에 하나씩 주어진다. 단어의 길이는 2와 100,000 사이이며, 모든 단어 길이의 합은 1,000,000을 넘지 않는다.

출력

첫째 줄에 좋은 단어의 수를 출력한다.


테스트 케이스

초록색 입력 / 검은색 출력

 

 


접근

문제를 이해하는데 좀 애를 먹은 것 같습니다.

문제를 해결하기 위해서는 AA BB 가 한쌍을 이뤄야 한다는 것을 알아야 합니다.

예를 들면 AABBBAA는 얼핏 보면 될 것 같지만 B하나가 쌍을 이루고 있지 않기 때문에 문제의 답으로 채택되지 않습니다.

 

따라서 스택을 이용하여 

현재 스택이 비어있으면 문자열의 값을 넣고

그게 아니면, 스텍이 비어있지 않고, 현재 스택의 PEEK()값과(최상단) 현재차례의 문자열이 일치하면 연결된 것으로 간주하여 현재 스텍의 PEEK()를 POP 합니다.

이 과정을 반복하여 문자열을 한 번씩 체크하고 난 후, 스택이 완전히 비어있다면 문제가 요하는 좋은 단어 이므로 COUNT+1 하면 문제가 해결됩니다. 


코드

import java.awt.desktop.SystemEventListener;
import java.io.*;
import java.math.*;
import java.util.*;

public class Main {

	/*
 	3986 problem 좋은 단어
	*/
	public static void main(String[] args) throws NumberFormatException, IOException {
		// TODO Auto-generated method stub
		
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
		Stack<Character> stack = new Stack<Character>();
		
		int num = Integer.parseInt(br.readLine());
		int result = 0;
		for(int i=0; i<num; i++) {
			String str = br.readLine();
			for(int j=0; j<str.length(); j++) {
				if(!stack.isEmpty() && stack.peek() == str.charAt(j)) {
					stack.pop();
				}
				else {
					stack.push(str.charAt(j));
				}
			}
			if(stack.isEmpty()) {
				result++;
			}
			stack.clear();
		}
		bw.write(String.valueOf(result));
		bw.flush();
		bw.close();
	}
}

주의

 

AA, BB 한쌍을 이루어야 합니다 따라서 AA는 되지만 AB는 정답으로 인정되지 않습니다.

반응형

'공부 정리 > 백준' 카테고리의 다른 글

[백준] 자바 2161 카드1  (0) 2021.06.24
[백준] 자바 1009 분산처리  (0) 2021.03.31
[백준] 자바 5397 키로거  (0) 2021.03.27
[백준] 자바 4949 균형잡힌 세상  (0) 2021.03.25
[백준] 자바 10799 쇠막대기  (0) 2021.03.22

댓글