tools/rotate-2d.hpp

• View this file on GitHub
• Last update: 2023-07-08 14:12:19+09:00
• Include: #include "tools/rotate-2d.hpp"

Depends on

• tools/base.hpp

Code

#pragma once

#include "./base.hpp"

#include <vector>

namespace matumoto {
  template <typename ArrayType>
  vector<ArrayType> rotate_cw(const vector<ArrayType> &s) {
    int h = s.size(), w = s[0].size();
    vector<ArrayType> res(w);
    for (int i = 0; i < w; i++) {
      for (int j = 0; j < h; j++) {
        res[i].push_back(s[h - 1 - j][i]);
      }
    }
    return res;
  }

  template <typename ArrayType>
  vector<ArrayType> rotate_ccw(const vector<ArrayType> &s) {
    int h = s.size(), w = s[0].size();
    vector<ArrayType> res(w);
    for (int i = 0; i < w; i++) {
      for (int j = 0; j < h; j++) {
        res[i].push_back(s[j][w - 1 - i]);
      }
    }
    return res;
  }
} // namespace matumoto

#line 2 "tools/rotate-2d.hpp"

#line 2 "tools/base.hpp"

namespace matumoto {
  using namespace std;
}
#line 4 "tools/rotate-2d.hpp"

#include <vector>

namespace matumoto {
  template <typename ArrayType>
  vector<ArrayType> rotate_cw(const vector<ArrayType> &s) {
    int h = s.size(), w = s[0].size();
    vector<ArrayType> res(w);
    for (int i = 0; i < w; i++) {
      for (int j = 0; j < h; j++) {
        res[i].push_back(s[h - 1 - j][i]);
      }
    }
    return res;
  }

  template <typename ArrayType>
  vector<ArrayType> rotate_ccw(const vector<ArrayType> &s) {
    int h = s.size(), w = s[0].size();
    vector<ArrayType> res(w);
    for (int i = 0; i < w; i++) {
      for (int j = 0; j < h; j++) {
        res[i].push_back(s[j][w - 1 - i]);
      }
    }
    return res;
  }
} // namespace matumoto

Back to top page