백준 16935 C++

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

16935번: 배열 돌리기 3

문제

---

크기가 N×M인 배열이 있을 때, 배열에 연산을 R번 적용하려고 한다. 연산은 총 6가지가 있다.

• 1번 연산은 배열을 상하 반전시키는 연산이다.
• 2번 연산은 배열을 좌우 반전시키는 연산이다.
• 3번 연산은 오른쪽으로 90도 회전시키는 연산이다.
• 4번 연산은 왼쪽으로 90도 회전시키는 연산이다.
• 5, 6번 연산을 수행하려면 배열을 크기가 N/2×M/2인 4개의 부분 배열로 나눠야 한다. 아래 그림은 크기가 6×8인 배열을 4개의 그룹으로 나눈 것이고, 1부터 4까지의 수로 나타냈다.
• 5번 연산은 1번 그룹의 부분 배열을 2번 그룹 위치로, 2번을 3번으로, 3번을 4번으로, 4번을 1번으로 이동시키는 연산이다. 
• 6번 연산은 1번 그룹의 부분 배열을 4번 그룹 위치로, 4번을 3번으로, 3번을 2번으로, 2번을 1번으로 이동시키는 연산이다.

- <<자세한 설명은 문제 링크에 나와있습니다.>>

 

 

입력

---

첫째 줄에 배열의 크기 N, M과 수행해야 하는 연산의 수 R이 주어진다.

둘째 줄부터 N개의 줄에 배열 A의 원소 Aij가 주어진다.

마지막 줄에는 수행해야 하는 연산이 주어진다. 연산은 공백으로 구분되어져 있고, 문제에서 설명한 연산 번호이며, 순서대로 적용시켜야 한다.

 

 

출력

---

입력으로 주어진 배열에 R개의 연산을 순서대로 수행한 결과를 출력한다.

 

 

접근법

---

1. 시뮬레이션 및 구현 문제이다 보니, 각 주어진 조건에 따라 구현을 성실히 해 나가면 된다.
2. 3,4 번 에서는 Col, Row 를 각각 뒤집어주어 resize 해주었으며,
3. 5,6 번은 구역을 나눠서, 배열을 계산해주었습니다.

 

 

#include <algorithm>
#include <iostream>
#include <iterator>
#include <sstream>
#include <vector>

using namespace std;

template <typename T> void print(T arr) {
  for (auto c : arr) {
    copy(begin(c), end(c), ostream_iterator<int>(cout, " "));
    cout << endl;
  }
}

void move_1(vector<vector<int>> &arr) {
  int row = arr.size();
  int col = arr[0].size();
  for (int i = 0; i < col; i++) {
    for (int j = 0; j < row / 2; j++) {
      swap(arr[j][i], arr[row - 1 - j][i]);
    }
  }
}

void move_2(vector<vector<int>> &arr) {
  int row = arr.size();
  int col = arr[0].size();
  for (int i = 0; i < row; i++) {
    for (int j = 0; j < col / 2; j++) {
      swap(arr[i][j], arr[i][col - 1 - j]);
    }
  }
}

void move_3(vector<vector<int>> &arr) {
  vector<vector<int>> tArr = arr;
  int row = arr.size();
  int col = arr[0].size();

  arr.resize(col);
  for (int i = 0; i < col; i++) {
    arr[i].resize(row);
  }

  for (int i = 0; i < row; i++) {
    for (int j = 0; j < col; j++) {
      arr[j][row - 1 - i] = tArr[i][j];
    }
  }
}

void move_4(vector<vector<int>> &arr) {
  vector<vector<int>> tArr = arr;
  int row = arr.size();
  int col = arr[0].size();

  arr.resize(col);
  for (int i = 0; i < col; i++) {
    arr[i].resize(row);
  }

  for (int i = 0; i < row; i++) {
    for (int j = 0; j < col; j++) {
      arr[col - 1 - j][i] = tArr[i][j];
    }
  }
}

void move_5(vector<vector<int>> &arr) {
  vector<vector<int>> tArr = arr;
  int N = arr.size() / 2;
  int M = arr[0].size() / 2;

  for (int i = 0; i < N; ++i)
    for (int j = 0; j < M; ++j) {
      arr[i][j + M] = tArr[i][j];
    }
  for (int i = 0; i < N; ++i)
    for (int j = 0; j < M; ++j) {
      arr[i + N][j + M] = tArr[i][j + M];
    }
  for (int i = 0; i < N; ++i)
    for (int j = 0; j < M; ++j) {
      arr[i + N][j] = tArr[i + N][j + M];
    }
  for (int i = 0; i < N; ++i)
    for (int j = 0; j < M; ++j) {
      arr[i][j] = tArr[i + N][j];
    }
}

void move_6(vector<vector<int>> &arr) {
  vector<vector<int>> tArr = arr;
  int N = arr.size() / 2;
  int M = arr[0].size() / 2;

  for (int i = 0; i < N; ++i)
    for (int j = 0; j < M; ++j) {
      arr[i + N][j + M] = tArr[i + N][j];
    }
  for (int i = 0; i < N; ++i)
    for (int j = 0; j < M; ++j) {
      arr[i][j + M] = tArr[i + N][j + M];
    }
  for (int i = 0; i < N; ++i)
    for (int j = 0; j < M; ++j) {
      arr[i + N][j] = tArr[i][j];
    }
  for (int i = 0; i < N; ++i)
    for (int j = 0; j < M; ++j) {
      arr[i][j] = tArr[i][j + M];
    }
}

int main() {
  int n, m, r;
  vector<vector<int>> arr;
  /*------------------ INPUT ------------------*/
  cin >> n >> m >> r;

  arr.resize(n);
  for (int i = 0; i < n; i++) {
    arr[i].resize(m);
  }
  for (int i = 0; i < n; i++) {
    for (int j = 0; j < m; j++) {
      cin >> arr[i][j];
    }
  }

  for (int i = 0; i < r; i++) {
    int T;
    cin >> T;
    switch (T) {
    case (1):
      move_1(arr);
      break;
    case (2):
      move_2(arr);
      break;
    case (3):
      move_3(arr);
      break;
    case (4):
      move_4(arr);
      break;
    case (5):
      move_5(arr);
      break;
    case (6):
      move_6(arr);
      break;
    default:
      break;
    }
  }

  print<vector<vector<int>>>(arr);
}