[BOJ] 백준 3986번 좋은 단어 (Java)

#3986 좋은 단어

난이도 : 실버 4

유형 : 자료구조/ stack/ 문자열 

3986번: 좋은 단어

▸ 문제

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

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

▸ 입력

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

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

▸ 출력

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

 

문제 풀이  

풀이는 문자열 메소드를 사용하는 방법과 스택을 활용하는 방법 두가지가 있다.

 

1. 문자열 replaceAll 사용

replaceAll (String regex, String replacement)

replaceAll 메소드는 문자열에 regex의 패턴이 있으면 해당 문자열을 replacement로 변환시켜준다.

 

line.replaceAll("AA","") : line문자열에 AA가 있다면 해당 문자열을 ""로 바꿔준다.

line.replaceAll("BB","") : line문자열에 BB가 있다면 해당 문자열을 ""로 바꿔준다.

import java.io.*;

public class Main{

	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		int n = Integer.parseInt(br.readLine());
		int cnt=0;
		for(int i=0; i<n; i++) {
			String line = br.readLine();
			String pattern1 = "AA";
			String pattern2 = "BB";
            
			while(true) {
				if(!line.contains(pattern1)&& !line.contains(pattern2)) break;
				if(line.contains(pattern1)) {
					line = line.replaceAll(pattern1, "");
				}
				if(line.contains(pattern2)) {
					line = line.replaceAll(pattern2, "");
				}
			}
			if(line.isEmpty()) {
				cnt++;
			}
		}
		System.out.println(cnt);
	}
}

 

2. Stack 사용

좋은 단어는 "AA"또는 "BB"가 나오면 제거해주면 되는 특징을 이용하여 이전에 들어왔던 문자와 현재 들어갈 문자가 같다면 스택에 담긴 이전 문자를 빼주는 연산을 진행해주면 된다. stack.peek() ==  line.charat(j) 

 

그렇게 stack에 아무것도 남지 않는다면 좋은 단어임을 알 수 있다. stack.size() == 0

import java.io.*;
import java.util.Stack;

public class Main {

	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		int n = Integer.parseInt(br.readLine());
		int cnt=0;
		for(int i=0; i<n; i++) {
			String line = br.readLine();
			Stack<Character> st = new Stack<>();
			
			for(int j=0; j<line.length(); j++) {
				if(st.size()>0 && st.peek() == line.charAt(j)) {
					st.pop();
				}else {
					st.push(line.charAt(j));
				}
			}
			if(st.size()==0) cnt++;
		}
		System.out.println(cnt);
	}
}