BOJ 23288 - 주사위 굴리기2(C++)

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

23288번: 주사위 굴리기 2

 

풀이

주사위 배열에 주사위 위치의 값 저장, 주사위 이동 시 배열 갱신

이동방향 dir[]={동, 남, 서, 북}

 

반복횟수 K만큼 진행

1. 주사위가 이동 방향으로 한칸 굴러감(굴러간 위치가 범위 밖이라면 반대방향으로 계산)

   - 주사위의 이동방향에 따라 주사위 배열의 값 수정

 

2. 주사위의 위치 (y, x)에 대해 점수 계산

   - 동서남북 방향으로 연속해서 이동할 수 있는(BFS) 칸의 수 * map[y][x] 값을 추가

 

3. 주사위의 아랫면, 주사위의 위치(y, x)를 바교 이동방향 결정

   - 주사위의 아랫면이 map[y][x]보다 크면 90도 회전 -> dir=(dir+1)%4

   - 주사위의 아랫면이 map[y][x]보다 작으면 90도 반시계 회전 -> dir=(4+dir-1)%4

 

#include <iostream>
#include <queue>
#include <cstring>
 
using namespace std;
	
int N, M, K;
bool visit[21][21];
int map[21][21];
int dice[] = { 1,6,4,3,5,2 };
int dy[] = {0,1,0,-1};
int dx[] = {1,0,-1,0};
 
int y = 1, x = 1;
int dir = 0;
 
void update_dice() {
	int up, down, left, right, front, back;
	up = dice[0];
	down = dice[1];
	left = dice[2];
	right = dice[3];
	front = dice[4];
	back = dice[5];
	// 동쪽으로 굴림
	if (dir == 0) {
		dice[0] = left;
		dice[1] = right;
		dice[2] = down;
		dice[3] = up;
	}
    //남쪽으로 굴림
	else if (dir == 1) {
		dice[0] = back;
		dice[1] = front;
		dice[4] = up;
		dice[5] = down;
	}
    //서쪽으로 굴람
	else if (dir == 2) {
		dice[0] = right;
		dice[1] = left;
		dice[2] = up;
		dice[3] = down;
	}
    //북쪽으로 굴림
	else {
		dice[0] = front;
		dice[1] = back;
		dice[4] = down;
		dice[5] = up;
	}
 
}
 
void update_dir() {
	//시계 방향 90도 회전
	if (dice[1] > map[y][x]) {
		dir = (dir + 1) % 4;
	}
    //반시계 방향 90도 회전
	else if (dice[1] < map[y][x]) {
		dir = (4 + dir - 1) % 4;
	}
}
 
// 주사위 이동
void move_dice() {
	int ny = y + dy[dir];
	int nx = x + dx[dir];
	//이동 불가 -> 반대 방향으로 변경 진행
	if (ny<1 || nx<1 || ny>N || nx>M) {
		dir = (dir + 2) % 4;
		ny = y + dy[dir];
		nx = x + dx[dir];
	}
    // 주사위를 굴려 이동 하므로 주사위의 상하좌우의 값이 변경
	update_dice();
	y = ny; x = nx;
}
 
//bfs 같은 값을 가진 칸의 수 구하기
int search_continuous() {
	int value = map[y][x];
	int cnt = 1;
 
	memset(visit, false, sizeof(visit));
	visit[y][x] = true;
	queue<pair<int, int>>q;
	q.push(make_pair(y, x));
 
	int ny, nx;
	while (!q.empty()) {
		int y = q.front().first;
		int x = q.front().second;
		q.pop();
 
		for (int d = 0; d < 4; d++) {
			ny = y + dy[d];
			nx = x + dx[d];
			if (ny<1 || nx<1 || ny>N || nx>M)continue;
			if (visit[ny][nx])continue;
			visit[ny][nx] = true;
			if (map[ny][nx] == value) {
				q.push(make_pair(ny, nx));
				cnt++;
			}
		}
	}
	return cnt;
}
 
 
int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(0);
 
	cin >> N >> M >> K;
 
	for (int i = 1; i <= N; i++) {
		for (int j = 1; j <= M; j++) {
			cin >> map[i][j];
		}
	}
	int score = 0;
	for (int i = 0; i < K; i++) {
		move_dice();
		//점수 추가
		score += map[y][x]*search_continuous();
		update_dir();
	}
	cout << score;
	return 0;
}