[Silver IV] 좋은 단어 - 3986

문제 링크

성능 요약

메모리: 21992 KB, 시간: 320 ms

분류

자료 구조, 스택

제출 일자

2024년 3월 12일 14:17:19

문제 설명

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

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

입력

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

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

출력

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

코드 (Stack 풀이)

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

public class Main {
	
	static int count=0; //좋은 단어 개수
	
	public static void main(String[] args) throws IOException{
		BufferedReader br =new BufferedReader(new InputStreamReader(System.in));
		int n = Integer.parseInt(br.readLine()); //단어 개수 입력받기.
		
		for(int i=0; i<n; i++) {
			String s = br.readLine();
			WordCheck(s); //좋은 단어 맞는지 확인
		}
		System.out.println(count); //좋은 단어 개수 출력
		
	}
	public static void WordCheck(String s) {
		if(s.length()%2==1) { //만약 단어 길이가 홀수면 무조건 1개가 남아서 좋은 단어 X
			return;
		}
		Stack<Character> stack = new Stack<>();
		stack.push(s.charAt(0)); //첫번째 단어 Stack에 넣기
		for(int i=1; i<s.length(); i++) {
			if(stack.size()>0 && stack.peek()==s.charAt(i)) { //만약 짝이 있다면?
				stack.pop(); 
			}
			else {
				stack.push(s.charAt(i)); //다른 단어일 경우 넣기.
			}
		}
		if(stack.isEmpty()) { //좋은 단어가 맞다면?
			count++; //개수 + 1
		}
	}
}

Stack을 사용해 해결하는 문제이다.
[스택]
[좋은 단어 문제 풀이 참고]

해결 방식

• 처음 단어 즉 0번째 단어를 Stack에 push()함으로서 시작한다.
• 1번부터 단어 길이 끝까지 Stack peek()와 비교해준다.
  • 만약 같은 단어일 경우 Stack.pop()해준다.
  • 다른 단어일 경우, 비교한 해당 단어를 push()해준다.

• [Stack push와 add의 차이?]
  • 처음 코드를 짤 때 add()를 사용했는데 push와의 차이점이 궁금해졌다.
  • push()는 stack에서 제공, add()는 List에서 제공하는 메서드이다.
  • push()의 리턴값은 <E>이고, add()의 리턴값은 boolean이다.