관리 메뉴

KoreanFoodie's Study

SW 역량테스트 - [백준] 구슬탈출2 문제 풀이/해답/코드 (C++ / JAVA) 본문

Data Structures, Algorithm/SW 역량테스트

SW 역량테스트 - [백준] 구슬탈출2 문제 풀이/해답/코드 (C++ / JAVA)

hashnut 2020. 9. 30. 10:53


SW 역량 테스트 준비를 위한 핵심 문제들을 다룹니다!

해답을 보기 전에 문제를 풀어보시거나, 설계를 하고 오시는 것을 추천드립니다. 

코드에 대한 설명은 주석을 참고해 주세요 :)

 


문제 링크 : www.acmicpc.net/problem/13460

해답 코드 : 

 

#include <iostream>
#include <algorithm>
#include <math.h>

using namespace std;

int N, M;
char map[10][10];

int ans;
bool b_hole; // true if blue goes down to hole

// Up, Down, Left, Right
// right, down, left, up
int dR[4] = {0, 1, 0, -1};
int dC[4] = {1, 0, -1, 0};

void copy_map(char temp[10][10], char ori[10][10]) {

	for (int i = 0; i < 10; i++) {
		for (int j = 0; j < 10; j++) {
			temp[i][j] = ori[i][j];
		}
	}
}

// initial cnt == 1
void dfs(int r_r, int r_c, int b_r, int b_c, int cnt) {
	
	if (cnt >= ans) {
		return;
	}

	char restore[10][10];
	copy_map(restore, map);

	int n_r_r;
	int n_r_c;
	int n_b_r;
	int n_b_c;

	char red_ori;
	char blue_ori;

	bool red_hole;
	bool blue_hole;
	bool red_stuck;

	for (int dir = 0; dir < 4; dir++) {

		n_r_r = r_r + dR[dir];
		n_r_c = r_c + dC[dir];
		n_b_r = b_r + dR[dir];
		n_b_c = b_c + dC[dir];

		// check if balls went to the hole
		red_hole = false;
		blue_hole = false;

		// if red is stuck with blue first
		red_stuck = false;

		// check whether it moved or not
		// move Red
		while (map[n_r_r][n_r_c] == '.') {
			n_r_r += dR[dir];
			n_r_c += dC[dir];
		}
		if (map[n_r_r][n_r_c] == '#') {
			swap(map[n_r_r - dR[dir]][n_r_c - dC[dir]], map[r_r][r_c]);
		}
		else if (map[n_r_r][n_r_c] == 'B') {
			red_stuck = true;
		}
		// hole
		else {
			map[r_r][r_c] = '.';
			red_hole = true;
		}

		// move Blue
		while (map[n_b_r][n_b_c] == '.') {
			n_b_r += dR[dir];
			n_b_c += dC[dir];
		}
		if ((map[n_b_r][n_b_c] == '#') || (map[n_b_r][n_b_c] == 'R')) {
			swap(map[n_b_r - dR[dir]][n_b_c - dC[dir]], map[b_r][b_c]);
		} // hole
		else {
			map[b_r][b_c] = '.';
			blue_hole = true;
		}

		if (red_stuck) {
			// move Red
			while (map[n_r_r][n_r_c] == '.') {
				n_r_r += dR[dir];
				n_r_c += dC[dir];
			}
			if ((map[n_r_r][n_r_c] == '#') || (map[n_r_r][n_r_c] == 'B')) {
				swap(map[n_r_r - dR[dir]][n_r_c - dC[dir]], map[r_r][r_c]);
			
			} // hole
			else {
				map[r_r][r_c] = '.';
				red_hole = true;
			}
		}
		
		// If blue goes down or get stuck, pass
		if (blue_hole) {
			copy_map(map, restore);
			continue;
		}
		
		else if (red_hole) {
			
			if (blue_hole) {
				copy_map(map, restore);
			}
			else {
				copy_map(map, restore);
				ans = min(ans, cnt);
				return;
			}

		}

		else if ((r_r == n_r_r - dR[dir] && r_c == n_r_c - dC[dir]
			&& b_r == n_b_r - dR[dir] && b_c == n_b_c - dC[dir])) {
			continue;
		}

		// red and blue is not in the hole
		else {

			dfs(n_r_r - dR[dir], n_r_c - dC[dir], n_b_r - dR[dir], n_b_c - dC[dir], cnt + 1);

			// restore map
			copy_map(map, restore);

		}

	}


}



int main() {

	//FILE* fp;
	//freopen_s(&fp, "sample_input.txt", "r", stdin);

	ans = 11;
	b_hole = false;

	cin >> N >> M;

	int r_r = 0, r_c = 0, b_r = 0, b_c = 0;

	for (int i = 0; i < N; i++) {

		for (int j = 0; j < M; j++) {
			cin >> map[i][j];
			if (map[i][j] == 'R') {
				r_r = i; r_c = j;
			}
			else if (map[i][j] == 'B') {
				b_r = i; b_c = j;
			}

		}
	}

	dfs(r_r, r_c, b_r, b_c, 1);

	if (ans == 11) ans = -1;

	cout << ans;

}

 

SW 역량테스트 준비 - [모의 SW 역량테스트] 풀이 / 코드 / 답안 (C++ / JAVA)

SW 역량테스트 준비 - C++ 코드, Java 코드

SW 역량테스트 준비 - 백준 알고리즘 문제

0 Comments
댓글쓰기 폼