백준 12904: A와 B
2022. 4. 25. 02:57ㆍ알고리즘/그리디 알고리즘
문제
수빈이는 A와 B로만 이루어진 영어 단어가 존재한다는 사실에 놀랐다. 대표적인 예로 AB (Abdominal의 약자), BAA (양의 울음 소리), AA (용암의 종류), ABBA (스웨덴 팝 그룹)이 있다.
이런 사실에 놀란 수빈이는 간단한 게임을 만들기로 했다. 두 문자열 S와 T가 주어졌을 때, S를 T로 바꾸는 게임이다. 문자열을 바꿀 때는 다음과 같은 두 가지 연산만 가능하다.
- 문자열의 뒤에 A를 추가한다.
- 문자열을 뒤집고 뒤에 B를 추가한다.
주어진 조건을 이용해서 S를 T로 만들 수 있는지 없는지 알아내는 프로그램을 작성하시오.
입력
첫째 줄에 S가 둘째 줄에 T가 주어진다. (1 ≤ S의 길이 ≤ 999, 2 ≤ T의 길이 ≤ 1000, S의 길이 < T의 길이)
내 제출
public class No12904 {
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
StringBuilder s = new StringBuilder(br.readLine());
StringBuilder t = new StringBuilder(br.readLine());
int k = t.length()-s.length();
for(int i =0; i<k; i++){
if(t.charAt(t.length()-1) == 'A'){
t = t.delete(t.length()-1, t.length());
}else{
t = t.delete(t.length()-1, t.length());
t = t.reverse();
}
}
if(t.toString().equals(s.toString())) System.out.println(1);
else System.out.println(0);
}
}
A를 B로 변환하는 경우의 수를 찾기보다, B를 하나씩 삭제하여 A가 될 수 있는지를 체크해야 한다.
비교군의 맨 마지막 인덱스가 "A"인 경우와 "B"인 경우 delete해주고 B인 경우에만 문자열을 뒤집는다.
모든 동작에 비교군의 인덱스가 한개씩 줄어들기 때문에, 두 문자열의 길이 차이 만큼 연산을 수행한다.
'알고리즘 > 그리디 알고리즘' 카테고리의 다른 글
백준: 4889 안정적인 문자열 [포포의 알고리즘] (0) | 2022.04.27 |
---|---|
백준 1758: 알바생 강호 (0) | 2022.04.26 |
백준 1343: 폴리오미노 (0) | 2022.04.23 |
백준 2170: 선 긋기 (0) | 2022.04.14 |
백준 19941: 햄버거 분배 (0) | 2022.04.08 |