[백준] BOJ 5002 도어맨 (실버 2)

문제

---

https://www.acmicpc.net/problem/5002

5002번: 도어맨

---

해설

---

1. 입력받은 문자열의 각 문자를 Deque에 넣는다.
2. 하나씩 꺼내며, 해당 문자가 M이면 man의 카운트를 1올리고, W이면 woman의 카운트를 1올린다.
3. 이때, 최대 기억할 수 있는 수를 넘어간다면 꺼낸 성별의 카운트를 다시 빼서 원상복구해준다.
4. 이후, Deque에서 다음 문자를 확인하고 현재 꺼낸 문자와 다르다면 해당 문자를 꺼내고
   이에 해당하는 성별의 카운트를 1올려준다.
   => 문제의 조건 : 두번째 사람을 먼저 입장시킬 수 있다.
5. 만약, Deque가 비어있거나 그 다음 사람도 성별이 같다면 현재 꺼낸 사람도 입장시킬 수 없으므로 카운트를 -1 하고 종료한다.
6. 최종적으로 입장한 man + woman의 카운트를 더해주면 입장한 사람의 수를 구할 수 있다.

---

코드

---

import java.io.*;
import java.util.ArrayDeque;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

        int X = Integer.parseInt(br.readLine());
        String input = br.readLine();

        ArrayDeque<Character> deque = new ArrayDeque<>();
        for (int i = 0; i < input.length(); i++) {
            deque.offerLast(input.charAt(i));
        }
        int man = 0;
        int woman = 0;
        while (!deque.isEmpty()) {
            char gender = deque.poll();
            if (gender == 'M') {
                man++;
                if (Math.abs(man - woman) > X && !deque.isEmpty()) {
                    char next = deque.peek();
                    man--;
                    if(next == 'W'){
                        deque.poll();
                        deque.offerFirst(gender);
                        woman++;
                    } else {
                        break;
                    }
                } else if (Math.abs(man - woman) > X && deque.isEmpty()) {
                    man--;
                    break;
                }
            } else {
                woman++;
                if (Math.abs(man - woman) > X && !deque.isEmpty()) {
                    char next = deque.peek();
                    woman--;
                    if(next == 'M'){
                        deque.poll();
                        deque.offerFirst(gender);
                        man++;
                    } else {
                        break;
                    }
                } else if (Math.abs(man - woman) > X && deque.isEmpty()) {
                    woman--;
                    break;
                }
            }

        }
        StringBuilder sb = new StringBuilder();
        sb.append(man + woman);
        bw.write(sb.toString());
        bw.flush();
        bw.close();
    }
}

---

결론

---

문제에서 요구하는 조건이 많지 않아 그냥 시키는대로 하면 풀리는 문제!

---