[Baekjoon] 5002_도어맨

Silver II

문제(출처: https://www.acmicpc.net/problem/5002)

---

< 도어맨 >

 

문제 풀이 

 

여자와 남자의 차이를 최소가 되도록 여자와 남자 수를 비교해 적은 성별을 먼저 입장시킨다. 만약 사람 수가 적은 성별을 먼저 입장시킬 수 없다면 정인이가 기억할 수 있는 차이 내에서 사람 수가 많은 성별을 입장시킨다.

 

 my solution (Java)

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

public class _5002_ { // 도어맨

	public static void main(String[] args) throws IOException {
		BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));

		int x = Integer.parseInt(bf.readLine());

		String str = bf.readLine();
		int w = 0, m = 0, idx1 = 0, idx2 = 1;

		while (Math.abs(m - w) <= x && x != 0) {
			if (idx1 == 0) {
				if (str.charAt(0) == 'M') {
					m += 1;
				} else {
					w += 1;
				}
				idx1 = idx2 + 1;
			}

			if (Math.abs(m - w) > x || (idx1 >= str.length() && idx2 >= str.length())) {
				break;
			}

			boolean flag = false;
			if (m > w) {
				if (idx1 < str.length() && w + 1 - m <= x && str.charAt(idx1) == 'W') {
					flag = true;
					w += 1;
					idx1 = Math.max(idx1, idx2) + 1;
				} else if (idx2 < str.length() && w + 1 - m <= x && str.charAt(idx2) == 'W') {
					flag = true;
					w += 1;
					idx2 = Math.max(idx1, idx2) + 1;
				} else if (idx1 < str.length() && m + 1 - w <= x && str.charAt(idx1) == 'M') {
					flag = true;
					m += 1;
					idx1 = Math.max(idx1, idx2) + 1;
				} else if (idx2 < str.length() && m + 1 - w <= x && str.charAt(idx2) == 'M') {
					flag = true;
					m += 1;
					idx2 = Math.max(idx1, idx2) + 1;
				}
			} else {
				if (idx1 < str.length() && m + 1 - w <= x && str.charAt(idx1) == 'M') {
					flag = true;
					m += 1;
					idx1 = Math.max(idx1, idx2) + 1;
				} else if (idx2 < str.length() && m + 1 - w <= x && str.charAt(idx2) == 'M') {
					flag = true;
					m += 1;
					idx2 = Math.max(idx1, idx2) + 1;
				} else if (idx1 < str.length() && w + 1 - m <= x && str.charAt(idx1) == 'W') {
					flag = true;
					w += 1;
					idx1 = Math.max(idx1, idx2) + 1;
				} else if (idx2 < str.length() && w + 1 - m <= x && str.charAt(idx2) == 'W') {
					flag = true;
					w += 1;
					idx2 = Math.max(idx1, idx2) + 1;
				}
			}
			if (!flag) {
				break;
			}
		}
		System.out.println(m + w);
	}
}

변수)
x : 기억할 수 있는 가장 큰 차이
str : 줄을 서 있는 순서
w, m : 입장한 여자, 남자 사람 수
idx1, idx2 : 줄을 서 있는 첫 번째, 두 번째 사람
flag : 입장 여부

 

기억할 수 있는 가장 큰 차이를 입력받는다. 줄을 서 있는 순서를 입력받아 입장한 여자와 남자 수 차이가 x이하일 때 다음 과정을 반복한다.

 

1) 제일 처음에 입장할 때는 여자인지 남자인지 상관없으므로 0번째 사람을 입장시킨다.

2) 입장한 여자와 남자의 수 차이가 x보다 크거나 줄을 서 있는 모든 사람을 다 입장시켰다면 종료한다.

3) 입장한 사람 중 남자 수가 더 많다면

3-1) 줄 서 있는 사람의 첫 번째 사람, 두 번째 사람 중에서 여자가 있다면 입장시킨다.

3-2) 여자가 없고 남자를 입장시켰을 때 입장한 여자와 남자의 차이 수가 x이하라면 남자를 입장시킨다.

4) 입장한 사람 중 여자 수가 더 많다면

4-1) 줄 서 있는 사람의 첫 번째 사람, 두 번째 사람 중에서 남자가 있다면 입장시킨다.

4-2) 남자가 없고 여자를 입장시켰을 때 입장한 여자와 남자의 차이 수가 x이하라면 여자를 입장시킨다.

5) 만약 아무도 입장시킬 수 없는 상황이라면 종료한다.

 

최종 입장한 사람 수인 m+w를 출력한다.

---

---