[BOJ] 17822. 원판 돌리기

* rotate에서는 x의 배수인 원판들을 시계/반시계 방향으로 k번 회전한다. 

* deque를 사용해서 시계방향은 한 원판에서 가장 뒤에 있는 요소를 가장 앞에 삽입한다. (pop_back -> push_front)

반시계방향은 가장 앞에 있는 요소를 가장 뒤에 삽입한다. (pop_front->push_back)

* deque에 저장된 값들을 다시 map에 넣는다. 

* removeNums()에서는 인접한 숫자들을 지운다. 만약, 전체 원판을 통틀어 지울 수가 없다면 평균값에 따라 원판의 값을 조절한다. 

* 각 좌표에서 상 하 좌 우를 살펴봤을 때 같은 수가 있으면 큐에 넣어준다. 만약, 전체 원판을 봤을 때 큐에 아무런 값도 없다면 지울 값이 없다는 것이다. 

* *평균값을 double로 구해줘야 한다. * *

* 지워진 숫자들은 -1로 표시해준다. 

 

 

#include <iostream> #include <vector> #include <deque> #include <cstring> #include <queue> using namespace std; int n, m, t; int map[51][51]; bool deleted[51][51]; bool flag[51]; int dx[4] = {0, 0, -1, 1}, dy[4] = {-1, 1, 0, 0}; typedef pair<int, int> pp; void print(int x) { for (int j = 1; j <= m; j++) { if (map[x][j] == -1) cout << 'x' << ' '; else { cout << map[x][j] << ' '; } } } void removeNums() { //인접한 숫자들을 지운다. queue<pp> q; for (int i = 1; i <= n; i++) { for (int j = 1; j <= m; j++) { if (map[i][j] == -1) continue; bool flag = false; for (int d = 0; d < 4; d++) { int nx = i + dx[d]; int ny = j + dy[d]; if (nx < 1 || nx > n || ny < 1 || ny > m) continue; if (map[i][j] == map[nx][ny]) { flag = true; q.push(pp(nx, ny)); } } if (j == 1 && map[i][1] == map[i][m]) { flag= true; q.push(pp(i, m)); } if (flag) q.push(pp(i, j)); } } if (q.empty()) { int sum = 0, cnt = 0; for (int i = 1; i <= n; i++) { for (int j = 1; j <= m; j++) { if (map[i][j] == -1) continue; sum += map[i][j]; cnt++; } } double avg = (double)((double)sum / (double)cnt); for (int i = 1; i <= n; i++) { for (int j = 1; j <= m; j++) { if (map[i][j] == -1) continue; if (map[i][j] < avg) map[i][j]++; else if (map[i][j] > avg) map[i][j]--; } } } else { while (!q.empty()) { int x = q.front().first; int y = q.front().second; q.pop(); map[x][y] = -1; } } } void rotate(int x, int d, int k) { deque<int> dq; for (int i = x; i <= n; i++) { if (i % x == 0) { for (int j = 1; j <= m; j++) dq.push_back(map[i][j]); //x의 배수인 원판 if (d == 0) { //시계방향 int cnt = k; while (cnt--) { dq.push_front(dq.back()); dq.pop_back(); } } else { int cnt = k; while (cnt--) { dq.push_back(dq.front()); dq.pop_front(); } } int idx = 1; while (!dq.empty()) { map[i][idx++] = dq.front(); dq.pop_front(); } } } removeNums(); } int main() { ios_base::sync_with_stdio(0); cin.tie(0); cin >> n >> m >> t; for (int i = 1; i <= n; i++) { for (int j = 1; j <= m; j++) { cin >> map[i][j]; } } while (t--) { int x, d, k; cin >> x >> d >> k; rotate(x, d, k); } int sum = 0; for (int i = 1; i <= n; i++) { for (int j = 1; j <= m; j++) { if (map[i][j] == -1) continue; sum += map[i][j]; } } cout << sum << '\n'; return 0; }